Add more types to the dmc space converter.

soroushmehr · Nov 24, 2021 · 8bc49fc · 8bc49fc
1 parent f8e5ed2
commit 8bc49fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jaxrl/wrappers/dmc_env.py b/jaxrl/wrappers/dmc_env.py
@@ -14,7 +14,7 @@
 
 
 def dmc_spec2gym_space(spec):
-    if isinstance(spec, OrderedDict):
+    if isinstance(spec, OrderedDict) or isinstance(spec, dict):
         spec = copy.copy(spec)
         for k, v in spec.items():
             spec[k] = dmc_spec2gym_space(v)