test type

Farama-Foundation · Aug 13, 2024 · dbddf3a · dbddf3a
1 parent a870455
commit dbddf3a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)