minor revision according to coderabbit

Author: ChiahsinChu

deepmd/pt/model/atomic_model/dp_atomic_model.py

@@ -152,6 +152,8 @@ def change_type_map(
             else None,
         )
         self.fitting_net.change_type_map(type_map=type_map)
+        # Reinitialize fitting to get correct sel_type
+        self.fitting_net.reinit_exclude(self.atom_exclude_types)
 
     def has_message_passing(self) -> bool:
         """Returns whether the atomic model has message passing."""

deepmd/tf/model/model.py

@@ -1011,7 +1011,7 @@ def deserialize(cls, data: dict, suffix: str = "") -> "Descriptor":
             sel_type = np.setdiff1d(
                 full_type_list, atom_exclude_types, assume_unique=True
             )
-            fitting_dict["sel_type"] = sel_type
+            fitting_dict["sel_type"] = sel_type.tolist()
         fitting = Fitting.deserialize(fitting_dict, suffix=suffix)
         # pass descriptor type embedding to model
         if descriptor.explicit_ntypes:

source/tests/pt/model/test_get_model.py

@@ -61,7 +61,10 @@ def test_model_attr(self) -> None:
             },
         )
         full_type_list = np.arange(len(atomic_model.type_map), dtype=int)
-        atom_exclude_types = np.setdiff1d(full_type_list, self.model.get_sel_type())
+        atom_exclude_types = np.setdiff1d(
+            full_type_list,
+            self.model.get_sel_type(),
+        ).tolist()
         self.assertEqual(atom_exclude_types, [1])
         self.assertEqual(atomic_model.atom_exclude_types, [1])
         self.assertEqual(atomic_model.pair_exclude_types, [[1, 2]])