import

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Mar 9, 2024 · d48b145 · d48b145
1 parent 54595aa
commit d48b145
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,