diff --git a/src/storage/model.cpp b/src/storage/model.cpp index a9c966a..45594d6 100644 --- a/src/storage/model.cpp +++ b/src/storage/model.cpp @@ -230,6 +230,7 @@ void define_sparse_model(py::module& m) { .def_property_readonly("backward_transition_matrix", &SparseModel::getBackwardTransitions, py::return_value_policy::reference, py::keep_alive<1, 0>(), "Backward transition matrix") .def("reduce_to_state_based_rewards", &SparseModel::reduceToStateBasedRewards) .def("__str__", getModelInfoPrinter("ParametricModel")) + .def("to_dot", [](SparseModel& model) { std::stringstream ss; model.writeDotToStream(ss); return ss.str(); }, "Write dot to a string") ; py::class_, std::shared_ptr>>(m, "SparseParametricDtmc", "pDTMC in sparse representation", modelRatFunc)