fix UP038 Use X | Y in isinstance call instead of (X, Y)

CederGroupHub · Sep 16, 2024 · 8ede53b · 8ede53b
1 parent 24d1657
commit 8ede53b
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/chgnet/model/dynamics.py b/chgnet/model/dynamics.py
@@ -499,7 +499,7 @@ def __init__(
         """
         self.ensemble = ensemble
         self.thermostat = thermostat
-        if isinstance(atoms, (Structure, Molecule)):
+        if isinstance(atoms, Structure | Molecule):
             atoms = AseAtomsAdaptor().get_atoms(atoms)
             # atoms = atoms.to_ase_atoms()
 

diff --git a/chgnet/model/model.py b/chgnet/model/model.py
@@ -622,7 +622,7 @@ def predict_graph(
                 m (Tensor) : magnetic moments of sites [num_atoms, 3] in Bohr
                     magneton mu_B
         """
-        if not isinstance(graph, (CrystalGraph, Sequence)):
+        if not isinstance(graph, CrystalGraph | Sequence):
             raise TypeError(
                 f"{type(graph)=} must be CrystalGraph or list of CrystalGraphs"
             )

diff --git a/pyproject.toml b/pyproject.toml
@@ -53,6 +53,7 @@ build-backend = "setuptools.build_meta"
 
 [tool.ruff]
 target-version = "py310"
+output-format = "concise"
 
 [tool.ruff.lint]
 select = ["ALL"]