From 2efc80f595f66084de041e97f797dc7857a39a0d Mon Sep 17 00:00:00 2001
From: Sebastian Junges <sebastian.junges@gmail.com>
Date: Tue, 26 May 2020 16:38:02 -0700
Subject: [PATCH] new statevaluation interface

---
 src/storage/valuation.cpp | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/storage/valuation.cpp b/src/storage/valuation.cpp
index aa8edef..dcd53af 100644
--- a/src/storage/valuation.cpp
+++ b/src/storage/valuation.cpp
@@ -9,9 +9,8 @@
 void define_statevaluation(py::module& m) {
 
     py::class_<storm::storage::sparse::StateValuations, std::shared_ptr<storm::storage::sparse::StateValuations>> statevaluation(m,"StateValuation", "Valuations for explicit states");
-    statevaluation.def("get_state", &storm::storage::sparse::StateValuations::getStateValuation, py::return_value_policy::reference_internal)
-            ;
-
+    statevaluation.def("get_boolean_value", &storm::storage::sparse::StateValuations::getBooleanValue, py::arg("state"), py::arg("variable"));
+    statevaluation.def("get_integer_value", &storm::storage::sparse::StateValuations::getIntegerValue, py::arg("state"), py::arg("variable"));
 }
 
 void define_simplevaluation(py::module& m) {