From bb6971965d90a31cae8055e8da127f06532bf48d Mon Sep 17 00:00:00 2001 From: PimLeerkes Date: Sun, 22 Dec 2024 12:28:46 +0100 Subject: [PATCH] small changes --- examples/monty_hall.py | 1 + stormvogel/model.py | 2 +- stormvogel/model_checking.py | 8 ++++---- 3 files changed, 6 insertions(+), 5 deletions(-) diff --git a/examples/monty_hall.py b/examples/monty_hall.py index e7e7520..473d29b 100644 --- a/examples/monty_hall.py +++ b/examples/monty_hall.py @@ -71,4 +71,5 @@ def create_monty_hall_mdp(): if __name__ == "__main__": # Print the resulting model in dot format. + print(create_monty_hall_mdp().to_dot()) diff --git a/stormvogel/model.py b/stormvogel/model.py index 5687751..13b2aec 100644 --- a/stormvogel/model.py +++ b/stormvogel/model.py @@ -92,7 +92,7 @@ def __init__( self.observation = None if name is None: - self.name = str(id) + self.name = str(id) # TODO Two states can have same name in some cases else: self.name = name diff --git a/stormvogel/model_checking.py b/stormvogel/model_checking.py index 0bc143d..c69b71c 100644 --- a/stormvogel/model_checking.py +++ b/stormvogel/model_checking.py @@ -49,9 +49,9 @@ def model_checking( if __name__ == "__main__": mdp = examples.monty_hall.create_monty_hall_mdp() - # rewardmodel = mdp.add_rewards("rewardmodel") - # rewardmodel.set_from_rewards_vector(list(range(67))) - # rewardmodel2 = mdp.add_rewards("rewardmodel2") - # rewardmodel2.set_from_rewards_vector(list(range(67))) + rewardmodel = mdp.add_rewards("rewardmodel") + rewardmodel.set_from_rewards_vector(list(range(67))) + rewardmodel2 = mdp.add_rewards("rewardmodel2") + rewardmodel2.set_from_rewards_vector(list(range(67))) print(model_checking(mdp)) # ,'R{"rewardmodel"}min=? [F "target" | "done"]'))