diff --git a/yt_idefix/data_structures.py b/yt_idefix/data_structures.py index 3a0f75d..4738cc8 100644 --- a/yt_idefix/data_structures.py +++ b/yt_idefix/data_structures.py @@ -14,6 +14,7 @@ import numpy as np import numpy.testing as npt +import yt from yt.data_objects.index_subobjects.stretched_grid import StretchedGrid from yt.data_objects.static_output import Dataset from yt.funcs import setdefaultattr @@ -423,17 +424,12 @@ def _setup_geometry(self) -> None: geom_str = from_disk def parse_geometry(geom: str): - import yt + if yt.version_info >= (4, 2): + from yt.geometry.api import Geometry # type: ignore [attr-defined] - if yt.version_info[:2] > (4, 1): - try: - from yt.geometry.api import Geometry # type: ignore [attr-defined] - - return Geometry(geom) - except ImportError: - pass - - return geom + return Geometry(geom) + else: + return geom self.geometry = parse_geometry(geom_str)