test debug

gnina · Jul 8, 2024 · 2412510 · 2412510
1 parent f22428c
commit 2412510
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/test/test_example_provider.py b/test/test_example_provider.py
@@ -437,7 +437,7 @@ def test_pytorch_iterdataset():
 
     lengths, centers, coords, types, radii, labels = next(m_iter)
 
-    assert list(center.shape) == [3]
+    assert list(centers.shape) == [BSIZE,3]
     np.testing.assert_allclose(coords[0,:lengths[0],:], coordinates.coords.tonumpy())
     np.testing.assert_allclose(types[0,:lengths[0]], coordinates.type_index.tonumpy())
     np.testing.assert_allclose(radii[0,:lengths[0]], coordinates.radii.tonumpy())
@@ -459,7 +459,6 @@ def test_pytorch_iterdataset():
     assert coords.shape[0] == BSIZE
     assert types.shape[0] == BSIZE
     assert radii.shape[0] == BSIZE
-    assert radii.shape[0] == BSIZE
     assert labels.shape[0] == BSIZE
 
     e.reset()