Small bug

mala-project · Apr 26, 2024 · 3cebb9d · 3cebb9d
1 parent bd68063
commit 3cebb9d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mala/network/trainer.py b/mala/network/trainer.py
@@ -672,7 +672,7 @@ def __prepare_to_train(self, optimizer_dict):
                 )
 
             if self.parameters_full.use_distributed_sampler_test:
-                if self.data.test_data_sets is not None:
+                if self.data.test_data_sets:
                     self.test_sampler = (
                         torch.utils.data.distributed.DistributedSampler(
                             self.data.test_data_sets[0],