Skip to content

Commit 8d155de

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

2 files changed

Lines changed: 4 additions & 2 deletions

File tree

deepmd/pt/train/validation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,8 +66,8 @@
6666

6767
if TYPE_CHECKING:
6868
from collections.abc import (
69-
Iterator,
7069
Callable,
70+
Iterator,
7171
)
7272

7373
from deepmd.utils.data import (

source/tests/pt/test_training.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1019,7 +1019,9 @@ def test_ema_checkpoint_cleanup_removes_future_steps(self) -> None:
10191019

10201020
@TRAINING_TEST_TIMEOUT
10211021
@patch("deepmd.pt.train.training.model_change_out_bias")
1022-
def test_ema_checkpoint_keeps_changed_out_bias(self, mocked_change_out_bias) -> None:
1022+
def test_ema_checkpoint_keeps_changed_out_bias(
1023+
self, mocked_change_out_bias
1024+
) -> None:
10231025
def change_out_bias(model, sample_func, _bias_adjust_mode):
10241026
model.set_out_bias(model.get_out_bias() + 1.0)
10251027
return model

0 commit comments

Comments
 (0)