gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-01 14:10:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix roundtrip test for ArrayConversion (#1397)

Browse Source
This commit is contained in:
Martin Schuck
2025-06-07 13:17:06 +02:00
committed by GitHub
parent 675945f105
commit de2c10ef1a
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/wrappers/test_array_conversion.py
View File

@@ -55,7 +55,8 @@ def xp_data_equivalence(data_1, data_2) -> bool:
xp_data_equivalence(o_1, o_2) for o_1, o_2 in zip(data_1, data_2)
)
elif is_array_api_obj(data_1):
return array_api_extra.isclose(data_1, data_2, atol=0.00001).all()
xp = array_namespace(data_1)
return xp.all(array_api_extra.isclose(data_1, data_2, atol=0.00001))
else:
return data_1 == data_2
else: