small fix unwrapped

BartekCupial · BartekCupial · commit 0e62efa59697 · 2024-11-02T06:39:01.000Z
diff --git a/nle/language_wrapper/wrappers/nle_language_wrapper.py b/nle/language_wrapper/wrappers/nle_language_wrapper.py
@@ -232,7 +232,7 @@ def __init__(self, env, use_language_action=True):
             use_language_action(bool): Use language action or discrete integer actions
         """
         super().__init__(env)
-        assert isinstance(env, NLE), "Only NLE environments are supported"
+        assert isinstance(env.unwrapped, NLE), "Only NLE environments are supported"
         missing_obsv_keys = self.REQUIRED_NLE_OBSV_KEYS.difference(
             env.observation_space.spaces.keys()
         )

Original file line number	Diff line number	Diff line change
`@@ -232,7 +232,7 @@ def __init__(self, env, use_language_action=True):`
`232`	`232`	`use_language_action(bool): Use language action or discrete integer actions`
`233`	`233`	`"""`
`234`	`234`	`super().__init__(env)`
`235`		`- assert isinstance(env, NLE), "Only NLE environments are supported"`
	`235`	`+ assert isinstance(env.unwrapped, NLE), "Only NLE environments are supported"`
`236`	`236`	`missing_obsv_keys = self.REQUIRED_NLE_OBSV_KEYS.difference(`
`237`	`237`	`env.observation_space.spaces.keys()`
`238`	`238`	`)`