Merge pull request #55 from BrainLesion/fix_unit_tests

fix imports
BrainLesion · Mar 31, 2024 · 129c759 · 129c759
2 parents 01d3cd9 + 422803d
commit 129c759
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