fix failing tests

jmbhughes · jmbhughes · commit 3b203c0f32d5 · 2024-07-25T14:42:41.000-06:00
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,

Original file line number	Diff line number	Diff line change
`@@ -45,7 +45,7 @@ def test_inversion_runs(tmp_path, inversion_mode, is_weighted):`
`45`	`45`	`detector_row_range=config["inversion"]["detector_row_range"],`
`46`	`46`	`)`
`47`	`47`
`48`		`- em_cube, prediction, scores, unconverged_rows = inversion.invert(`
	`48`	`+ em_cube, prediction, scores, unconverged_rows, niter = inversion.invert(`
`49`	`49`	`overlappogram,`
`50`	`50`	`config["model"],`
`51`	`51`	`3E-5,`
Original file line number	Diff line number	Diff line change
`@@ -34,7 +34,7 @@ def test_create_spectrally_pure_images(tmp_path):`
`34`	`34`	`detector_row_range=config["inversion"]["detector_row_range"],`
`35`	`35`	`)`
`36`	`36`
`37`		`- em_cube, prediction, scores, unconverged_rows = inversion.invert(`
	`37`	`+ em_cube, prediction, scores, unconverged_rows, _ = inversion.invert(`
`38`	`38`	`overlappogram,`
`39`	`39`	`config["model"],`
`40`	`40`	`3E-5,`