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

for more information, see https://pre-commit.ci
deepmodeling · Apr 7, 2024 · 919dad1 · 919dad1
1 parent 9f0979d
commit 919dad1
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/deepmd/pt/model/atomic_model/dp_atomic_model.py b/deepmd/pt/model/atomic_model/dp_atomic_model.py
@@ -178,7 +178,7 @@ def forward_atomic(
 
     def get_out_bias(self) -> torch.Tensor:
         return self.out_bias
-        
+
     def compute_or_load_stat(
         self,
         sampled_func,

diff --git a/deepmd/pt/model/atomic_model/linear_atomic_model.py b/deepmd/pt/model/atomic_model/linear_atomic_model.py
@@ -98,7 +98,7 @@ def mixed_types(self) -> bool:
 
     def get_out_bias(self) -> torch.Tensor:
         return self.out_bias
-        
+
     def get_rcut(self) -> float:
         """Get the cut-off radius."""
         return max(self.get_model_rcuts())

diff --git a/deepmd/pt/model/atomic_model/pairtab_atomic_model.py b/deepmd/pt/model/atomic_model/pairtab_atomic_model.py
@@ -138,7 +138,7 @@ def fitting_output_def(self) -> FittingOutputDef:
 
     def get_out_bias(self) -> torch.Tensor:
         return self.out_bias
-        
+
     def get_rcut(self) -> float:
         return self.rcut
 

diff --git a/deepmd/pt/model/model/make_model.py b/deepmd/pt/model/model/make_model.py
@@ -172,7 +172,7 @@ def forward_common(
             model_predict = self.output_type_cast(model_predict, input_prec)
             return model_predict
 
-        def get_out_bias(self)-> torch.Tensor:
+        def get_out_bias(self) -> torch.Tensor:
             return self.atomic_model.get_out_bias()
 
         def change_out_bias(