[python] CostStack : use overload_base_get_item_for_map

This commit requires stack-of-tasks/eigenpy#507 to be merged.
Simple-Robotics · Sep 23, 2024 · f8146a3 · f8146a3
1 parent 6d0638c
commit f8146a3
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/bindings/python/src/modelling/expose-cost-stack.cpp b/bindings/python/src/modelling/expose-cost-stack.cpp
@@ -68,7 +68,8 @@ void exposeCostStack() {
             .def(CopyableVisitor<CostStack>())
             .def(PolymorphicMultiBaseVisitor<CostAbstract>());
 #if ALIGATOR_EIGENPY_HAS_MAP_SUPPORT
-    eigenpy::GenericMapVisitor<CostMap, true>::expose("CostMap");
+    eigenpy::GenericMapVisitor<CostMap, true>::expose(
+        "CostMap", eigenpy::overload_base_get_item_for_map<CostMap>());
 #endif
   }
 
@@ -79,7 +80,8 @@ void exposeCostStack() {
             "CostStackData", "Data struct for CostStack.", bp::no_init)
             .def_readonly("sub_cost_data", &CostStackData::sub_cost_data);
 #if ALIGATOR_EIGENPY_HAS_MAP_SUPPORT
-    eigenpy::GenericMapVisitor<CostStackData::DataMap, true>::expose("DataMap");
+    eigenpy::GenericMapVisitor<CostStackData::DataMap, true>::expose(
+        "CostMap", eigenpy::overload_base_get_item_for_map<CostMap>());
 #endif
   }
 }