diff --git a/examples/shields/rl/15_train_eval_tune.py b/examples/shields/rl/15_train_eval_tune.py index 720d11c..808b431 100644 --- a/examples/shields/rl/15_train_eval_tune.py +++ b/examples/shields/rl/15_train_eval_tune.py @@ -139,7 +139,7 @@ def main(): ppo(args) - + ray.shutdown() if __name__ == '__main__': main() \ No newline at end of file diff --git a/examples/shields/rl/shieldhandlers.py b/examples/shields/rl/shieldhandlers.py index eda1768..5ca4b54 100644 --- a/examples/shields/rl/shieldhandlers.py +++ b/examples/shields/rl/shieldhandlers.py @@ -69,7 +69,7 @@ class MiniGridShieldHandler(ShieldHandler): action_dictionary[state_valuation] = actions_to_be_executed - stormpy.shields.export_shield(model, shield, "Grid.shield") + # stormpy.shields.export_shield(model, shield, "Grid.shield") return action_dictionary