diff --git a/gtsam/inference/Factor.h b/gtsam/inference/Factor.h index e357a9c88..f073c4975 100644 --- a/gtsam/inference/Factor.h +++ b/gtsam/inference/Factor.h @@ -25,6 +25,7 @@ #include #endif #include +#include #include #include diff --git a/python/CMakeLists.txt b/python/CMakeLists.txt index 404f5ecc0..674f17d3f 100644 --- a/python/CMakeLists.txt +++ b/python/CMakeLists.txt @@ -257,7 +257,7 @@ if(GTSAM_UNSTABLE_BUILD_PYTHON) COMMAND ${CMAKE_COMMAND} -E env "PYTHONPATH=${GTSAM_PYTHON_BUILD_DIRECTORY}/$ENV{PYTHONPATH}" - pybind11-stubgen -o . --enum-class-locations \"KernelFunctionType|NoiseFormat:gtsam.gtsam\" --enum-class-locations \"OrderingType:gtsam.gtsam.Ordering\" gtsam_unstable + pybind11-stubgen -o . --enum-class-locations \"KernelFunctionType|NoiseFormat:gtsam.gtsam\" --enum-class-locations \"OrderingType:gtsam.gtsam.Ordering\" --numpy-array-use-type-var gtsam_unstable DEPENDS ${GTSAM_PYTHON_DEPENDENCIES} ${GTSAM_PYTHON_TEST_FILES} ${GTSAM_PYTHON_UNSTABLE_TARGET} WORKING_DIRECTORY "${GTSAM_PYTHON_BUILD_DIRECTORY}/" ) @@ -284,7 +284,7 @@ add_custom_target( COMMAND ${CMAKE_COMMAND} -E env "PYTHONPATH=${GTSAM_PYTHON_BUILD_DIRECTORY}/$ENV{PYTHONPATH}" - pybind11-stubgen -o . --enum-class-locations \"KernelFunctionType|NoiseFormat:gtsam.gtsam\" --enum-class-locations \"OrderingType:gtsam.gtsam.Ordering\" gtsam + pybind11-stubgen -o . --enum-class-locations \"KernelFunctionType|NoiseFormat:gtsam.gtsam\" --enum-class-locations \"OrderingType:gtsam.gtsam.Ordering\" --numpy-array-use-type-var gtsam DEPENDS ${GTSAM_PYTHON_DEPENDENCIES} ${GTSAM_PYTHON_TEST_FILES} ${GTSAM_PYTHON_TARGET} WORKING_DIRECTORY "${GTSAM_PYTHON_BUILD_DIRECTORY}/" ) diff --git a/python/gtsam/gtsam.tpl b/python/gtsam/gtsam.tpl index c72a216a2..6876b4ab4 100644 --- a/python/gtsam/gtsam.tpl +++ b/python/gtsam/gtsam.tpl @@ -39,12 +39,12 @@ namespace py = pybind11; {module_def} {{ m_.doc() = "pybind11 wrapper of {module_name}"; +// Specializations for STL classes +#include "python/gtsam/specializations/{module_name}.h" + {submodules_init} {wrapped_namespace} -// Specializations for STL classes -#include "python/gtsam/specializations/{module_name}.h" - }}