fix failing tests

ECCCO-mission · Jul 25, 2024 · 3b203c0 · 3b203c0
1 parent 9fff569
commit 3b203c0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_inversion.py b/tests/test_inversion.py
@@ -45,7 +45,7 @@ def test_inversion_runs(tmp_path, inversion_mode, is_weighted):
         detector_row_range=config["inversion"]["detector_row_range"],
     )
 
-    em_cube, prediction, scores, unconverged_rows = inversion.invert(
+    em_cube, prediction, scores, unconverged_rows, niter = inversion.invert(
         overlappogram,
         config["model"],
         3E-5,

diff --git a/tests/test_spectral.py b/tests/test_spectral.py
@@ -34,7 +34,7 @@ def test_create_spectrally_pure_images(tmp_path):
         detector_row_range=config["inversion"]["detector_row_range"],
     )
 
-    em_cube, prediction, scores, unconverged_rows = inversion.invert(
+    em_cube, prediction, scores, unconverged_rows, _ = inversion.invert(
         overlappogram,
         config["model"],
         3E-5,