Make the global optimizer work in python3

ced9f6f4 · Davis King · affaead6 · ced9f6f4
Commit ced9f6f4 authored Dec 19, 2017 by Davis King
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

tools/python/src/global_optimization.cpp tools/python/src/global_optimization.cpp +4 -1

No files found.
--- a/tools/python/src/global_optimization.cpp
+++ b/tools/python/src/global_optimization.cpp
@@ -49,7 +49,10 @@ boost::python::list mat_to_list (

 size_t num_function_arguments(object f)
 {
-    return boost::python::extract<std::size_t>(f.attr("func_code").attr("co_argcount"));
+    if (hasattr(f,"func_code"))
+        return boost::python::extract<std::size_t>(f.attr("func_code").attr("co_argcount"));
+    else
+        return boost::python::extract<std::size_t>(f.attr("__code__").attr("co_argcount"));
 }

 double call_func(object f, const matrix<double,0,1>& args)