narrow types

wanghan-iapcm · Dec 10, 2024 · 099a163 · 099a163
1 parent 7c14d45
commit 099a163
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dpdata/system.py b/dpdata/system.py
@@ -1049,6 +1049,7 @@ def remove_atom_names(self, atom_names: str | list[str]):
             atom_idx = self.data["atom_types"] == idx
             removed_atom_idx.append(atom_idx)
         picked_atom_idx = ~np.any(removed_atom_idx, axis=0)
+        assert not isinstance(picked_atom_idx, np.bool)
         new_sys = self.pick_atom_idx(picked_atom_idx)
         # let's remove atom_names
         # firstly, rearrange atom_names and put these atom_names in the end