From 27999af02ef59f6b4081eb3a134c7dd6a63d6121 Mon Sep 17 00:00:00 2001
From: SumGuo <guo0808@emails.bjut.edu.cn>
Date: Mon, 6 Jan 2025 09:48:50 +0800
Subject: [PATCH] check import

---
 deepmd/pt/utils/stat.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/deepmd/pt/utils/stat.py b/deepmd/pt/utils/stat.py
index ceb790c792..3c47b1c268 100644
--- 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,
 )