fix E721 Do not compare types, use isinstance()

deepmodeling · Aug 22, 2023 · 45659ad · 45659ad
1 parent ef4e450
commit 45659ad
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/dpdata/abacus/md.py b/dpdata/abacus/md.py
@@ -206,7 +206,7 @@ def get_frame(fname):
     data["energies"] = energy
     data["forces"] = force
     data["virials"] = stress
-    if type(data["virials"]) != np.ndarray:
+    if not isinstance(data["virials"], np.ndarray):
         del data["virials"]
     data["orig"] = np.zeros(3)
 

diff --git a/dpdata/md/rdf.py b/dpdata/md/rdf.py
@@ -62,9 +62,9 @@ def _compute_rdf_1frame(box, posis, atype, sel_type=[None, None], max_r=5, nbins
         sel_type[0] = all_types
     if sel_type[1] is None:
         sel_type[1] = all_types
-    if type(sel_type[0]) is not list:
+    if not isinstance(sel_type[0], list):
         sel_type[0] = [sel_type[0]]
-    if type(sel_type[1]) is not list:
+    if not isinstance(sel_type[1], list):
         sel_type[1] = [sel_type[1]]
     natoms = len(posis)
     import ase.neighborlist