diff --git a/tests/data/_IsmrmrdRawTestData.py b/tests/data/_IsmrmrdRawTestData.py index 0ba8fa2d..efeff6ed 100644 --- a/tests/data/_IsmrmrdRawTestData.py +++ b/tests/data/_IsmrmrdRawTestData.py @@ -9,6 +9,7 @@ from einops import repeat from mrpro.data import SpatialDimension from mrpro.phantoms import EllipsePhantom + from tests import RandomGenerator ISMRMRD_TRAJECTORY_TYPE = ( diff --git a/tests/utils/test_split_idx.py b/tests/utils/test_split_idx.py index 1cd206b8..30501b9a 100644 --- a/tests/utils/test_split_idx.py +++ b/tests/utils/test_split_idx.py @@ -4,6 +4,7 @@ import torch from einops import repeat from mrpro.utils import split_idx + from tests import RandomGenerator