adding BaseCalculator to type check

Author: bowen-bd

chgnet/model/dynamics.py

@@ -10,7 +10,12 @@
 
 import numpy as np
 from ase import Atoms, units
-from ase.calculators.calculator import Calculator, all_changes, all_properties
+from ase.calculators.calculator import (
+    BaseCalculator,
+    Calculator,
+    all_changes,
+    all_properties,
+)
 from ase.md.npt import NPT
 from ase.md.nptberendsen import Inhomogeneous_NPTBerendsen, NPTBerendsen, NVTBerendsen
 from ase.md.velocitydistribution import MaxwellBoltzmannDistribution, Stationary
@@ -530,7 +535,7 @@ def __init__(
             Stationary(atoms)
 
         self.atoms = atoms
-        if isinstance(model, Calculator):
+        if isinstance(model, Calculator | BaseCalculator):
             self.atoms.calc = model
         else:
             self.atoms.calc = CHGNetCalculator(