Better benchmark imports.

trexfeathers · Nov 21, 2022 · e543a3c · e543a3c
1 parent c3fa502
commit e543a3c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/benchmarks/benchmarks/ci/esmf_regridder.py b/benchmarks/benchmarks/ci/esmf_regridder.py
@@ -9,14 +9,14 @@
 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,
     MeshToGridESMFRegridder,
 )
 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
@@ -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