fix: precommit

deepmodeling · Mar 8, 2024 · e4086fb · e4086fb
1 parent 4f63e95
commit e4086fb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/deepmd/dpmodel/fitting/polarizability_fitting.py b/deepmd/dpmodel/fitting/polarizability_fitting.py
@@ -183,8 +183,8 @@ def __getitem__(self, key):
         if key in ["constant_matrix"]:
             return self.constant_matrix
         else:
-            super().__getitem__(key)
-
+            return super().__getitem__(key)
+            
     def serialize(self) -> dict:
         data = super().serialize()
         data["type"] = "polar"

diff --git a/deepmd/pt/model/task/polarizability.py b/deepmd/pt/model/task/polarizability.py
@@ -162,7 +162,7 @@ def __getitem__(self, key):
         if key in ["constant_matrix"]:
             return self.constant_matrix
         else:
-            super().__getitem__(key)
+            return super().__getitem__(key)
 
     def serialize(self) -> dict:
         data = super().serialize()