From 5a01fb35bda57735aaa7d6177facd747e179317d Mon Sep 17 00:00:00 2001 From: Sebastian Junges Date: Mon, 14 Aug 2017 15:37:18 +0200 Subject: [PATCH] add missing reduce to state based rewards for parametric models --- src/storage/model.cpp | 1 + 1 file changed, 1 insertion(+) 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")) ;