diff --git a/src/logic/formulae.cpp b/src/logic/formulae.cpp index 8b80dea..05981a3 100644 --- a/src/logic/formulae.cpp +++ b/src/logic/formulae.cpp @@ -72,7 +72,9 @@ void define_formulae(py::module& m) { py::class_>(m, "TimeOperator", "The time operator", operatorFormula); py::class_>(m, "LongRunAvarageOperator", "Long run average operator", operatorFormula); py::class_>(m, "ProbabilityOperator", "Probability operator", operatorFormula); - py::class_>(m, "RewardOperator", "Reward operator", operatorFormula); + py::class_>(m, "RewardOperator", "Reward operator", operatorFormula) + .def("has_reward_name", &storm::logic::RewardOperatorFormula::hasRewardModelName) + .def_property_readonly("reward_name", &storm::logic::RewardOperatorFormula::getRewardModelName); py::class_> binaryStateFormula(m, "BinaryStateFormula", "State formula with two operands", stateFormula); py::class_>(m, "BooleanBinaryStateFormula", "Boolean binary state formula", binaryStateFormula); }