move get_model_def_script to make_base_atomic_model

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Feb 27, 2024 · d3ef0a4 · d3ef0a4
1 parent fbd1300
commit d3ef0a4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/deepmd/dpmodel/atomic_model/make_base_atomic_model.py b/deepmd/dpmodel/atomic_model/make_base_atomic_model.py
@@ -160,6 +160,10 @@ def do_grad_(self, var_name: str, base: str) -> bool:
                 return self.fitting_output_def()[var_name].c_differentiable
             return self.fitting_output_def()[var_name].r_differentiable
 
+        def get_model_def_script(self) -> str:
+            # TODO: implement this method; saved to model
+            raise NotImplementedError
+
     setattr(BAM, fwd_method_name, BAM.fwd)
     delattr(BAM, "fwd")
 

diff --git a/deepmd/dpmodel/model/base_model.py b/deepmd/dpmodel/model/base_model.py
@@ -156,7 +156,3 @@ class BaseModel(make_base_model()):
     deepmd.dpmodel.model.base_model.BaseBaseModel
         Backend-independent BaseModel class.
     """
-
-    def get_model_def_script(self) -> str:
-        # TODO: implement this method; saved to model
-        raise NotImplementedError