[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
deepmodeling · Nov 2, 2024 · 6e93586 · 6e93586
1 parent 9ac5204
commit 6e93586
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/deepmd/dpmodel/model/base_model.py b/deepmd/dpmodel/model/base_model.py
@@ -190,7 +190,7 @@ def update_sel(
                 model_type = local_jdata.get("fitting", {}).get("type", "ener")
             cls = cls.get_class_by_type(model_type)
             return cls.update_sel(train_data, type_map, local_jdata)
-        
+
         def enable_compression(
             self,
             table_extrapolate: float = 5,
@@ -202,8 +202,6 @@ def enable_compression(
 
             Parameters
             ----------
-            min_nbor_dist
-                The nearest distance between atoms
             table_extrapolate
                 The scale of model extrapolation
             table_stride_1

diff --git a/deepmd/pt/model/atomic_model/dp_atomic_model.py b/deepmd/pt/model/atomic_model/dp_atomic_model.py
@@ -159,7 +159,7 @@ def deserialize(cls, data) -> "DPAtomicModel":
         data["fitting"] = fitting_obj
         obj = super().deserialize(data)
         return obj
-    
+
     def enable_compression(
         self,
         min_nbor_dist: float,

diff --git a/deepmd/pt/model/model/make_model.py b/deepmd/pt/model/model/make_model.py
@@ -97,7 +97,7 @@ def model_output_type(self) -> list[str]:
                 if vv.category == OutputVariableCategory.OUT.value:
                     vars.append(kk)
             return vars
-        
+
         def enable_compression(
             self,
             table_extrapolate: float = 5,