diff --git a/test/utils.py b/test/utils.py index 0855f37..2402f98 100644 --- a/test/utils.py +++ b/test/utils.py @@ -76,7 +76,7 @@ def run_env_determinism(env_id: str, seed: int, iterations: int = None, wrappers for i, time_step in enumerate(zip(*traj1, *traj2)): obs1, rwd1, term1, trunc1, ac1, obs2, rwd2, term2, trunc2, ac2 = time_step assert np.allclose( - obs1, obs2), f"Observations [{i}] {obs1} and {obs2} do not match." + obs1, obs2), f"Observations [{i}] {obs1} ({obs1.shape}) and {obs2} ({obs2.shape}) do not match." assert np.array_equal( ac1, ac2), f"Actions [{i}] {ac1} and {ac2} do not match." assert np.array_equal(