Skip to content

Commit ae84b7e

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 15ebc08 commit ae84b7e

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

source/tests/pt/model/test_saveload_dpa1.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,9 +71,11 @@ def setUp(self) -> None:
7171
drop_last=False,
7272
pin_memory=True,
7373
)
74+
7475
def cycle_iterator(iterable):
7576
while True:
7677
yield from iterable
78+
7779
with torch.device("cpu"):
7880
self.training_data = cycle_iterator(self.training_dataloader)
7981
self.loss = EnergyStdLoss(**self.config["loss"])

0 commit comments

Comments
 (0)