Fix test due to rounding error

DLR-RM · Jan 25, 2025 · db6066f · db6066f
1 parent 1e32fe1
commit db6066f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_vec_normalize.py b/tests/test_vec_normalize.py
@@ -315,7 +315,7 @@ def test_get_original():
         assert not np.array_equal(orig_obs, obs)
         assert not np.array_equal(orig_rewards, rewards)
         np.testing.assert_allclose(venv.normalize_obs(orig_obs), obs)
-        np.testing.assert_allclose(venv.normalize_reward(orig_rewards), rewards)
+        np.testing.assert_allclose(venv.normalize_reward(orig_rewards), rewards, atol=1e-6)
 
 
 def test_get_original_dict():