diff --git a/src/storage/model.cpp b/src/storage/model.cpp index 6415e8a..6bb90cb 100644 --- a/src/storage/model.cpp +++ b/src/storage/model.cpp @@ -157,6 +157,7 @@ void define_model(py::module& m) { .def_property_readonly("reward_models", [](Model const& model) {return model.getRewardModels(); }, "Reward models") .def_property_readonly("transition_matrix", &getTransitionMatrix, py::return_value_policy::reference, py::keep_alive<1, 0>(), "Transition matrix") .def_property_readonly("backward_transition_matrix", &getBackwardTransitionMatrix, py::return_value_policy::reference, py::keep_alive<1, 0>(), "Backward transition matrix") + .def("reduce_to_state_based_rewards", &Model::reduceToStateBasedRewards) .def("__str__", getModelInfoPrinter("ParametricModel")) ;