Skip to content

Commit

Permalink
Merge branch 'load-data' of https://github.com/caic99/deepmd-kit into…
Browse files Browse the repository at this point in the history
… load-data
  • Loading branch information
caic99 committed Dec 19, 2024
2 parents 4954f06 + 7bf77cb commit 10cbd06
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion deepmd/pt/utils/dataloader.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,7 @@ def construct_dataset(system):
with Pool(os.cpu_count()) as pool:
self.systems = pool.map(construct_dataset, systems)
else:
self.systems = [None] * len(systems) # type: ignore
self.systems = [None] * len(systems) # type: ignore
if dist.is_initialized():
dist.broadcast_object_list(self.systems)
assert self.systems[-1] is not None
Expand Down

0 comments on commit 10cbd06

Please sign in to comment.