amend

pytorch · Sep 4, 2024 · ac5e5ef · ac5e5ef
1 parent b807bb7
commit ac5e5ef
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/torchrl/envs/libs/vmas.py b/torchrl/envs/libs/vmas.py
@@ -96,10 +96,9 @@ def _vmas_to_torchrl_spec_transform(
         )
     elif isinstance(spec, gym_spaces.Dict):
         spec_out = {}
-        for k in spec.keys():
-            key = k
+        for key in spec.keys():
             spec_out[key] = _vmas_to_torchrl_spec_transform(
-                spec[k],
+                spec[key],
                 device=device,
                 categorical_action_encoding=categorical_action_encoding,
             )