diff --git a/src/impl/vamp/bindings/python/environment.cc b/src/impl/vamp/bindings/python/environment.cc index ae296b84..aa041b5d 100644 --- a/src/impl/vamp/bindings/python/environment.cc +++ b/src/impl/vamp/bindings/python/environment.cc @@ -1,5 +1,6 @@ #include +#include #include #include #include @@ -162,7 +163,14 @@ void vamp::binding::init_environment(nanobind::module_ &pymodule) .def( "attach", [](vc::Environment &e, const vc::Attachment &a) { e.attachments.emplace(a); }) - .def("detach", [](vc::Environment &e) { e.attachments.reset(); }); + .def("detach", [](vc::Environment &e) { e.attachments.reset(); }) + .def_ro("spheres", &vc::Environment::spheres) + .def_ro("cuboids", &vc::Environment::cuboids) + .def_ro("z_aligned_cuboids", &vc::Environment::z_aligned_cuboids) + .def_ro("capsules", &vc::Environment::capsules) + .def_ro("z_aligned_capsules", &vc::Environment::z_aligned_capsules) + .def_ro("heightfields", &vc::Environment::heightfields) + .def_ro("pointclouds", &vc::Environment::pointclouds); pymodule.def( "filter_pointcloud",