Fix building

salvacarrion · Jun 12, 2019 · b4517f7 · b4517f7
1 parent 9753505
commit b4517f7
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 9 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -489,13 +489,14 @@ SET(TESTS
 
 
 # [TESTS] Generate a test executable
+SET(TEST_LIB_NAME "${PROJECT_NAME}_test")
 add_subdirectory(third_party/catch2)
-add_executable("${PROJECT_NAME}_test" ${SOURCES} ${TESTS})
-target_link_libraries("${PROJECT_NAME}_test" Catch2::Catch2)
-
+add_executable(${TEST_LIB_NAME} ${SOURCES} ${TESTS})
+target_link_libraries(${TEST_LIB_NAME} Catch2::Catch2)
+target_link_libraries(${TEST_LIB_NAME} Threads::Threads)
 
 # [BINDING] Create pyhton binding
+SET(C_LIB_NAME "_C")
 add_subdirectory(third_party/pybind11)
-pybind11_add_module(_C ${SOURCES} "${SOURCE_DIR}/binding.cpp")
-
+pybind11_add_module(${C_LIB_NAME} ${SOURCES} "${SOURCE_DIR}/binding.cpp")
 
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,9 @@
 setuptools==40.8.0
 numpy==1.16.2
 tqdm==4.32.1
-protobuf==3.7.1
-onnx==1.4.1
-onnxconverter-common==1.4.2
-onnxruntime
+six==1.12.0
+
+#protobuf==3.7.1
+#onnx==1.4.1
+#onnxconverter-common==1.4.2
+#onnxruntime