From a4201fe485194810eafec5d20d4a6828c1bf0579 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 21 Mar 2024 08:19:49 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- deepmd/pt/model/atomic_model/linear_atomic_model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/deepmd/pt/model/atomic_model/linear_atomic_model.py b/deepmd/pt/model/atomic_model/linear_atomic_model.py index 2cf1cc556c..ae8b246233 100644 --- a/deepmd/pt/model/atomic_model/linear_atomic_model.py +++ b/deepmd/pt/model/atomic_model/linear_atomic_model.py @@ -96,17 +96,17 @@ def mixed_types(self) -> bool: def get_rcut(self) -> float: """Get the cut-off radius.""" return max(self.get_model_rcuts()) - + @torch.jit.export def get_type_map(self) -> List[str]: """Get the type map.""" return self.type_map - + @torch.jit.export def get_model_rcuts(self) -> List[float]: """Get the cut-off radius for each individual models.""" return [model.get_rcut() for model in self.models] - + @torch.jit.export def get_sel(self) -> List[int]: return [max([model.get_nsel() for model in self.models])]