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

for more information, see https://pre-commit.ci
deepmodeling · Mar 25, 2024 · 858eed4 · 858eed4
1 parent fbf6d86
commit 858eed4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/deepmd/pt/utils/utils.py b/deepmd/pt/utils/utils.py
@@ -117,7 +117,9 @@ def dict_to_device(sample_dict):
         if isinstance(sample_dict[key], list):
             sample_dict[key] = [item.to(DEVICE) for item in sample_dict[key]]
         if isinstance(sample_dict[key], np.float32):
-            sample_dict[key] = torch.ones(1, dtype=torch.float64, device=DEVICE) * sample_dict[key]
+            sample_dict[key] = (
+                torch.ones(1, dtype=torch.float64, device=DEVICE) * sample_dict[key]
+            )
         else:
             if sample_dict[key] is not None:
                 sample_dict[key] = sample_dict[key].to(DEVICE)