check import

deepmodeling · Jan 6, 2025 · 27999af · 27999af
1 parent aa2633d
commit 27999af
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/deepmd/pt/utils/stat.py b/deepmd/pt/utils/stat.py
@@ -1,5 +1,7 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
 import logging
+import numpy as np
+import torch
 from collections import (
     defaultdict,
 )
@@ -8,7 +10,9 @@
     Optional,
     Union,
 )
-
+from deepmd.dpmodel.output_def import (
+    FittingOutputDef,
+)
 from deepmd.pt.utils import (
     AtomExcludeMask,
 )