update test

geoelements · Jul 3, 2024 · 86b4dbb · 86b4dbb
1 parent 2bcd131
commit 86b4dbb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_particle_data_loader.py b/test/test_particle_data_loader.py
@@ -154,7 +154,7 @@ def mock_get_rank():
 
 def test_get_data_loader_distributed(dummy_npz_data, mock_distributed_env):
     data_path, _ = dummy_npz_data
-    loader = get_data_loader(data_path, is_distributed=True)
+    loader = get_data_loader(data_path, use_dist=True)
 
     assert isinstance(loader, DataLoader)
     assert isinstance(loader.sampler, DistributedSampler)