From e7c09070707db90ad6ca1d8b26e839de9d05cd12 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 26 Jan 2024 17:24:54 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- source/tests/pt/test_jit.py | 3 ++- source/tests/pt/test_smooth.py | 1 - 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/source/tests/pt/test_jit.py b/source/tests/pt/test_jit.py index f833a20a2e..d9d88ddbc1 100644 --- a/source/tests/pt/test_jit.py +++ b/source/tests/pt/test_jit.py @@ -9,7 +9,9 @@ from pathlib import ( Path, ) + import torch + from deepmd.pt.entrypoints.main import ( get_trainer, ) @@ -26,7 +28,6 @@ class JITTest: - def test_jit(self): trainer = get_trainer(deepcopy(self.config)) trainer.run() diff --git a/source/tests/pt/test_smooth.py b/source/tests/pt/test_smooth.py index bacb40e87f..2e3bf61d10 100644 --- a/source/tests/pt/test_smooth.py +++ b/source/tests/pt/test_smooth.py @@ -26,7 +26,6 @@ class SmoothTest: - def test( self, ):