From d48b145a99a4d7e1513c6b02fa877a2d823d80cd Mon Sep 17 00:00:00 2001
From: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Date: Fri, 8 Mar 2024 20:30:41 -0500
Subject: [PATCH] import

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
---
 deepmd/dpmodel/model/make_model.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/deepmd/dpmodel/model/make_model.py b/deepmd/dpmodel/model/make_model.py
index 36790bd36f..e0f7201886 100644
--- a/deepmd/dpmodel/model/make_model.py
+++ b/deepmd/dpmodel/model/make_model.py
@@ -30,6 +30,7 @@
     OutputVariableCategory,
     OutputVariableOperation,
     check_operation_applied,
+    FittingOutputDef,
 )
 from deepmd.dpmodel.utils import (
     build_neighbor_list,