Skip to content

Commit 47632b0

Browse files
committed
Attempt to fix tests that fail on GitHub but succeed locally
1 parent 47a5497 commit 47632b0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_vectorized.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -374,8 +374,8 @@ def test_before_after_broadcast_tracking_equal_ares_ea():
374374

375375
for i in range(3):
376376
for j in range(10):
377-
assert torch.all(broadcast_outgoing._mu[i, j] == outgoing._mu[0])
378-
assert torch.all(broadcast_outgoing._cov[i, j] == outgoing._cov[0])
377+
assert torch.allclose(broadcast_outgoing._mu[i, j], outgoing._mu[0])
378+
assert torch.allclose(broadcast_outgoing._cov[i, j], outgoing._cov[0])
379379

380380

381381
def test_broadcast_customtransfermap():

0 commit comments

Comments
 (0)