From 3648aa6f7657c654cd06dd451a4de86be789269d Mon Sep 17 00:00:00 2001
From: Sebastian Junges <sebastian.junges@rwth-aachen.de>
Date: Tue, 2 Oct 2018 18:52:45 +0200
Subject: [PATCH] model to dot string

---
 src/storage/model.cpp | 1 +
 1 file changed, 1 insertion(+)

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<RationalFunction>::getBackwardTransitions, py::return_value_policy::reference, py::keep_alive<1, 0>(), "Backward transition matrix")
         .def("reduce_to_state_based_rewards", &SparseModel<RationalFunction>::reduceToStateBasedRewards)
         .def("__str__", getModelInfoPrinter<RationalFunction>("ParametricModel"))
+        .def("to_dot", [](SparseModel<RationalFunction>& model) { std::stringstream ss; model.writeDotToStream(ss); return ss.str(); }, "Write dot to a string")
     ;
 
     py::class_<SparseDtmc<RationalFunction>, std::shared_ptr<SparseDtmc<RationalFunction>>>(m, "SparseParametricDtmc", "pDTMC in sparse representation", modelRatFunc)