Skip to content

Commit d0a2316

Browse files
committed
remove check
1 parent 055cc63 commit d0a2316

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/xminigrid/manual_control.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
import xminigrid
1010

1111
from .environment import Environment, EnvParamsT
12+
from .envs.xland import XLandEnvParams
1213
from .rendering.text_render import print_ruleset
1314
from .types import EnvCarryT
1415
from .wrappers import GymAutoResetWrapper
@@ -162,9 +163,10 @@ def close(self) -> None:
162163

163164
if "XLand" in args.env_id:
164165
bench = xminigrid.load_benchmark(args.benchmark_id)
165-
env_params = env_params.replace(ruleset=bench.get_ruleset(args.ruleset_id))
166-
assert hasattr(env_params, "ruleset")
167-
print_ruleset(env_params.ruleset)
166+
ruleset = bench.get_ruleset(args.ruleset_id)
167+
168+
env_params = env_params.replace(ruleset=ruleset)
169+
print_ruleset(ruleset)
168170
print()
169171

170172
control = ManualControl(env=env, env_params=env_params, agent_view=args.agent_view)

0 commit comments

Comments
 (0)