change "setup.py"

Author: BigNoobWasTaken

.gitignore

@@ -5,3 +5,4 @@ temp/
 __pycache__/
 *.egg-info/*
 dist/
+*so

CMakeLists.txt

@@ -3,20 +3,16 @@ find_package(Python COMPONENTS Interpreter Development)
 if (${Python_FOUND})
 
 project(TensorArray_Python VERSION 0.1.0 LANGUAGES C CXX)
-set(CMAKE_ARCHIVE_OUTPUT_DIRECTORY ${CMAKE_CURRENT_LIST_DIR}/lib)
-set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${CMAKE_CURRENT_LIST_DIR}/src/tensor_array/core)
-set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_CURRENT_LIST_DIR}/bin)
 
 include(CTest)
 enable_testing()
 
 add_subdirectory("third_party/pybind11" EXCLUDE_FROM_ALL)
 add_subdirectory("tensor-array-repo/Tensor-Array" EXCLUDE_FROM_ALL)
-add_library(tensor2 SHARED src/tensor_array/_core/tensor_bind.cc)
+pybind11_add_module(tensor2 SHARED src/tensor_array/_core/tensor_bind.cc)
 
 target_include_directories(
     tensor2
-    PRIVATE "${CMAKE_CURRENT_LIST_DIR}/third_party/pybind11/include"
     PRIVATE "${CMAKE_CURRENT_LIST_DIR}/tensor-array-repo/Tensor-Array/src"
     PRIVATE ${Python_INCLUDE_DIRS}
 )
@@ -29,6 +25,5 @@ set(CPACK_PROJECT_NAME ${PROJECT_NAME})
 set(CPACK_PROJECT_VERSION ${PROJECT_VERSION})
 include(CPack)
 
-install(TARGETS tensor2 LIBRARY DESTINATION /src/tensor_array/core ARCHIVE DESTINATION lib)
 endif()
 

setup.py

@@ -130,11 +130,11 @@ def main():
 
     setup(
         name = "TensorArray",
-        version = "0.0.1a02",
+        version = "0.0.1a2",
         description = "A machine learning package",
         long_description=long_description,
         ext_modules=[
-            CMakeExtension("tensor_array")
+            CMakeExtension("tensor2")
         ],
         authors = "TensorArray-Creators",
         url= "https://github.com/Tensor-Array/Tensor-Array-Python",