Skip to content

Commit 583e2a1

Browse files
authored
[BugFix] Fix lib tests (#2122)
1 parent d8a60df commit 583e2a1

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

test/test_libs.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -2824,7 +2824,7 @@ def preproc(data):
28242824
preproc,
28252825
num_workers=max(1, os.cpu_count() - 4),
28262826
num_chunks=1000,
2827-
mp_start_method="fork",
2827+
# mp_start_method="fork",
28282828
pbar=True,
28292829
dest=tmpdir,
28302830
num_frames=100,
@@ -3000,7 +3000,7 @@ def fn(data: TensorDict):
30003000
CloudpickleWrapper(fn),
30013001
num_workers=max(1, os.cpu_count() - 2),
30023002
num_chunks=500,
3003-
mp_start_method="fork",
3003+
# mp_start_method="fork",
30043004
dest=tmpdir,
30053005
)
30063006
dataset = ReplayBuffer(storage=new_storage)

torchrl/envs/libs/jumanji.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -67,8 +67,8 @@ def _jumanji_to_torchrl_spec_transform(
6767
dtype = numpy_to_torch_dtype_dict[spec.dtype]
6868
return BoundedTensorSpec(
6969
shape=shape,
70-
low=np.asarray(spec.low),
71-
high=np.asarray(spec.high),
70+
low=np.asarray(spec.minimum),
71+
high=np.asarray(spec.maximum),
7272
dtype=dtype,
7373
device=device,
7474
)

0 commit comments

Comments
 (0)