fix: UTs

deepmodeling · Mar 6, 2024 · edd9a10 · edd9a10
1 parent a0d7caf
commit edd9a10
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/pt/model/atomic_model/linear_atomic_model.py b/deepmd/pt/model/atomic_model/linear_atomic_model.py
@@ -191,7 +191,7 @@ def forward_atomic(
         ener_list = []
 
         for i, model in enumerate(self.models):
-            mapping = self.mapping_list[i] if mapping is None else mapping
+            mapping = self.mapping_list[i]
             ener_list.append(
                 model.forward_atomic(
                     extended_coord,