From 422803d346083e89b4b2f0a9e9c99be4651f674b Mon Sep 17 00:00:00 2001 From: neuronflow Date: Sun, 31 Mar 2024 12:01:12 +0200 Subject: [PATCH] fix imports Signed-off-by: neuronflow --- unit_tests/test_full.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/unit_tests/test_full.py b/unit_tests/test_full.py index 536c70d..4b4acfd 100644 --- a/unit_tests/test_full.py +++ b/unit_tests/test_full.py @@ -1,8 +1,12 @@ -import os, tempfile, shutil, yaml +import os +import tempfile from pathlib import Path +import yaml + from ereg.cli.run import main -from ereg.registration import RegistrationClass, registration_function +from ereg.functional import registration_function +from ereg.registration import RegistrationClass from ereg.utils.io import read_image_and_cast_to_32bit_float