Skip to content

Commit 6c65bba

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

1 file changed

Lines changed: 6 additions & 3 deletions

File tree

deepmd/pt/train/training.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,9 @@
22
import functools
33
import logging
44
import time
5+
from collections.abc import (
6+
Iterable,
7+
)
58
from copy import (
69
deepcopy,
710
)
@@ -11,7 +14,6 @@
1114
from typing import (
1215
Any,
1316
)
14-
from collections.abc import Iterable
1517

1618
import numpy as np
1719
import torch
@@ -166,8 +168,9 @@ def cycle_iterator(iterable: Iterable):
166168
Args:
167169
iterable (Iterable): The iterable to cycle through.
168170
169-
Yields:
170-
Any: The next item from the iterable, cycling back to the beginning when the end is reached.
171+
Yields
172+
------
173+
Any: The next item from the iterable, cycling back to the beginning when the end is reached.
171174
"""
172175
while True:
173176
yield from iterable

0 commit comments

Comments
 (0)