fix imports

Signed-off-by: neuronflow <[email protected]>
BrainLesion · Mar 31, 2024 · 422803d · 422803d
1 parent 01d3cd9
commit 422803d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git 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