fixing test

LSSTDESC · Feb 25, 2022 · 05da1bb · 05da1bb
1 parent 4768049
commit 05da1bb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_group_sampling.py b/tests/test_group_sampling.py
@@ -43,5 +43,5 @@ def test_group_sampling():
     rel_diff2 = (batch_mean - test_batch_mean) / test_batch_mean
     rel_diff3 = (batch_std - test_batch_std) / test_batch_std
     assert np.all(rel_diff1 <= 0.1)
-    assert rel_diff2 <= 0.1
-    assert rel_diff3 <= 0.1
+    assert np.all(rel_diff2 <= 0.1)
+    assert np.all(rel_diff3 <= 0.1)