File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change 9
9
import xminigrid
10
10
11
11
from .environment import Environment , EnvParamsT
12
+ from .envs .xland import XLandEnvParams
12
13
from .rendering .text_render import print_ruleset
13
14
from .types import EnvCarryT
14
15
from .wrappers import GymAutoResetWrapper
@@ -162,9 +163,10 @@ def close(self) -> None:
162
163
163
164
if "XLand" in args .env_id :
164
165
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 )
168
170
print ()
169
171
170
172
control = ManualControl (env = env , env_params = env_params , agent_view = args .agent_view )
You can’t perform that action at this time.
0 commit comments