test debug

gnina · Jul 8, 2024 · c355145 · c355145
1 parent da2aa72
commit c355145
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/test_example_provider.py b/test/test_example_provider.py
@@ -449,6 +449,7 @@ def test_pytorch_iterdataset():
     np.testing.assert_allclose(labels[0,-1], 0.162643)
 
     # ensure it works with more than 1 worker
+    m.examples.reset()
     torch_loader = torch.utils.data.DataLoader(
         m, batch_size=None, num_workers=2)
     iterator = iter(torch_loader)