diff --git a/benchmarks/benchmarks/ci/esmf_regridder.py b/benchmarks/benchmarks/ci/esmf_regridder.py index 3a0efc01..4059b22b 100644 --- a/benchmarks/benchmarks/ci/esmf_regridder.py +++ b/benchmarks/benchmarks/ci/esmf_regridder.py @@ -9,7 +9,6 @@ from iris.cube import Cube from iris.experimental.ugrid import PARSE_UGRID_ON_LOAD -from benchmarks import disable_repeat_between_setup from esmf_regrid.esmf_regridder import GridInfo from esmf_regrid.experimental.unstructured_scheme import ( GridToMeshESMFRegridder, @@ -17,6 +16,7 @@ ) from esmf_regrid.schemes import ESMFAreaWeightedRegridder +from .. import disable_repeat_between_setup from ..generate_data import _grid_cube, _gridlike_mesh_cube diff --git a/benchmarks/benchmarks/long/esmf_regridder.py b/benchmarks/benchmarks/long/esmf_regridder.py index 1a655587..7baff174 100644 --- a/benchmarks/benchmarks/long/esmf_regridder.py +++ b/benchmarks/benchmarks/long/esmf_regridder.py @@ -8,13 +8,14 @@ import iris from iris.cube import Cube -from benchmarks import disable_repeat_between_setup, skip_benchmark from esmf_regrid.experimental.io import load_regridder, save_regridder from esmf_regrid.experimental.unstructured_scheme import ( GridToMeshESMFRegridder, MeshToGridESMFRegridder, ) from esmf_regrid.schemes import ESMFAreaWeightedRegridder + +from .. import disable_repeat_between_setup, skip_benchmark from ..generate_data import _grid_cube, _gridlike_mesh_cube