[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
deepmodeling · Jun 4, 2024 · 8ee1125 · 8ee1125
1 parent 001d21c
commit 8ee1125
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/tests/pt/model/test_se_atten_v2.py b/source/tests/pt/model/test_se_atten_v2.py
@@ -138,4 +138,4 @@ def test_jit(
             )
             dd0.se_atten.mean = torch.tensor(davg, dtype=dtype, device=env.DEVICE)
             dd0.se_atten.dstd = torch.tensor(dstd, dtype=dtype, device=env.DEVICE)
-            _ = torch.jit.script(dd0)
+            _ = torch.jit.script(dd0)