diff --git a/src/core/core.cpp b/src/core/core.cpp index 2f218a9..e966019 100644 --- a/src/core/core.cpp +++ b/src/core/core.cpp @@ -133,6 +133,7 @@ void define_build(py::module& m) { .def(py::init(), "Initialise without formulae", py::arg("build_all_reward_models")=true, py::arg("build_all_labels")=true) .def_property_readonly("preserved_label_names", &storm::builder::BuilderOptions::getLabelNames, "Labels preserved") .def("set_build_state_valuations", &storm::builder::BuilderOptions::setBuildStateValuations, "Build state valuations", py::arg("new_value")=true) + .def("set_build_observation_valuations", &storm::builder::BuilderOptions::setBuildObservationValuations, "Build observation valuations", py::arg("new_value")=true) .def("set_build_with_choice_origins", &storm::builder::BuilderOptions::setBuildChoiceOrigins, "Build choice origins", py::arg("new_value")=true) .def("set_add_out_of_bounds_state", &storm::builder::BuilderOptions::setAddOutOfBoundsState, "Build with out of bounds state", py::arg("new_value")=true) .def("set_add_overlapping_guards_label", &storm::builder::BuilderOptions::setAddOverlappingGuardsLabel, "Build with overlapping guards state labeled", py::arg("new_value")=true) diff --git a/src/storage/model.cpp b/src/storage/model.cpp index ee97b69..152c140 100644 --- a/src/storage/model.cpp +++ b/src/storage/model.cpp @@ -217,6 +217,8 @@ void define_sparse_model(py::module& m) { .def("get_observation", &SparsePomdp::getObservation, py::arg("state")) .def_property_readonly("observations", &SparsePomdp::getObservations) .def_property_readonly("nr_observations", &SparsePomdp::getNrObservations) + .def("has_observation_valuations", &SparsePomdp::hasObservationValuations) + .def_property_readonly("observation_valuations", &SparsePomdp::getObservationValuations) ; py::class_, std::shared_ptr>>(m, "SparseCtmc", "CTMC in sparse representation", model) .def(py::init>(), py::arg("other_model"))