diff --git a/tests/test_wrappers.py b/tests/test_wrappers.py index 91d4abf..df2ded4 100644 --- a/tests/test_wrappers.py +++ b/tests/test_wrappers.py @@ -87,6 +87,6 @@ def test_mo_record_ep_statistic(): np.testing.assert_allclose(info["episode"]["dr"], [7.71538, -2.9109], rtol=0, atol=1e-2) # 0 * 0.97**0 + 0 * 0.97**1 + 8.2 * 0.97**2 == 7.71538 # -1 * 0.97**0 + -1 * 0.97**1 + -1 * 0.97**2 == -2.9109 - assert isinstance(info["episode"]["l"], np.int32) + assert isinstance(info["episode"]["l"], int) assert info["episode"]["l"] == 3 assert isinstance(info["episode"]["t"], float)