add check

dunnolab · Mar 4, 2024 · 055cc63 · 055cc63
1 parent 92a54ff
commit 055cc63
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/xminigrid/manual_control.py b/src/xminigrid/manual_control.py
@@ -163,6 +163,7 @@ def close(self) -> None:
     if "XLand" in args.env_id:
         bench = xminigrid.load_benchmark(args.benchmark_id)
         env_params = env_params.replace(ruleset=bench.get_ruleset(args.ruleset_id))
+        assert hasattr(env_params, "ruleset")
         print_ruleset(env_params.ruleset)
         print()