diff --git a/pyiron_atomistics/calphy/job.py b/pyiron_atomistics/calphy/job.py index 9c32eb19a..8ea30c071 100644 --- a/pyiron_atomistics/calphy/job.py +++ b/pyiron_atomistics/calphy/job.py @@ -27,7 +27,13 @@ from calphy import Alchemy, Calculation, Liquid, Solid from calphy import __version__ as calphy_version from calphy.routines import routine_alchemy, routine_fe, routine_pscale, routine_ts - from pyscal.trajectory import Trajectory as PyscalTrajectory + + # both trajectory classes behave the same, so we can use either depending + # on the env + try: + from pyscal.trajectory import Trajectory as PyscalTrajectory + except ImportError: + from pyscal3.trajectory import Trajectory as PyscalTrajectory __author__ = "Sarath Menon" __copyright__ = (