Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
amend
Browse files Browse the repository at this point in the history
vmoens committed Feb 15, 2024
1 parent e79ef5e commit 2d01867
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions test/test_env.py
Original file line number Diff line number Diff line change
@@ -2539,7 +2539,7 @@ def make_env(seed, device=device):
return env

serial_env = SerialEnv(
2, [functools.partial(make_env, seed=0), functools.partial(make_env, seed=seed)]
2, [functools.partial(make_env, seed=0), functools.partial(make_env, seed=seed)], device=device
)
r_serial = serial_env.rollout(10, policy)

@@ -2549,7 +2549,7 @@ def make_env(seed, device=device):
torch.testing.assert_close(g, g_serial)

p_env = ParallelEnv(
2, [functools.partial(make_env, seed=0), functools.partial(make_env, seed=seed)]
2, [functools.partial(make_env, seed=0), functools.partial(make_env, seed=seed)], device=device,
)
try:
r_parallel = p_env.rollout(10, policy)

0 comments on commit 2d01867

Please sign in to comment.