[BugFix] Fix lib tests (pytorch#2122)

ai4co · Apr 25, 2024 · 583e2a1 · 583e2a1
1 parent d8a60df
commit 583e2a1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/test/test_libs.py b/test/test_libs.py
@@ -2824,7 +2824,7 @@ def preproc(data):
             preproc,
             num_workers=max(1, os.cpu_count() - 4),
             num_chunks=1000,
-            mp_start_method="fork",
+            # mp_start_method="fork",
             pbar=True,
             dest=tmpdir,
             num_frames=100,
@@ -3000,7 +3000,7 @@ def fn(data: TensorDict):
             CloudpickleWrapper(fn),
             num_workers=max(1, os.cpu_count() - 2),
             num_chunks=500,
-            mp_start_method="fork",
+            # mp_start_method="fork",
             dest=tmpdir,
         )
         dataset = ReplayBuffer(storage=new_storage)

diff --git a/torchrl/envs/libs/jumanji.py b/torchrl/envs/libs/jumanji.py
@@ -67,8 +67,8 @@ def _jumanji_to_torchrl_spec_transform(
             dtype = numpy_to_torch_dtype_dict[spec.dtype]
         return BoundedTensorSpec(
             shape=shape,
-            low=np.asarray(spec.low),
-            high=np.asarray(spec.high),
+            low=np.asarray(spec.minimum),
+            high=np.asarray(spec.maximum),
             dtype=dtype,
             device=device,
         )