diff --git a/deepmd/pt/utils/dataloader.py b/deepmd/pt/utils/dataloader.py index a46df3fe4a..961a8ed9c4 100644 --- a/deepmd/pt/utils/dataloader.py +++ b/deepmd/pt/utils/dataloader.py @@ -92,6 +92,7 @@ def construct_dataset(system): system=system, type_map=type_map, ) + self.systems: list[DeepmdDataSetForLoader] = [] global_rank = dist.get_rank() if dist.is_initialized() else 0 if global_rank == 0: