Update CMakeLists

2024-04-10 12:55:23 +02:00
parent a660c66c01
commit 1e88222775
2 changed files with 3 additions and 2 deletions
--- a/pyclfs/CMakeLists.txt
+++ b/pyclfs/CMakeLists.txt
@@ -2,7 +2,7 @@ include_directories(
    ${Python3_INCLUDE_DIRS}
    ${TORCH_INCLUDE_DIRS}
    ${PyClassifiers_SOURCE_DIR}/lib/json/include
-    /usr/local/include
+    ${Bayesnet_INCLUDE_DIRS}
 )
 add_library(PyClassifiers ODTE.cc STree.cc SVC.cc RandomForest.cc XGBoost.cc PyClassifier.cc PyWrap.cc)
 target_link_libraries(PyClassifiers ${Python3_LIBRARIES} "${TORCH_LIBRARIES}" ${LIBTORCH_PYTHON} Boost::boost Boost::python Boost::numpy)