stability improvement and also deprecation warning avoidance

sibirrer · Jan 26, 2024 · 0dd976d · 0dd976d
1 parent 9f0365b
commit 0dd976d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lenstronomy/ImSim/image_linear_solve.py b/lenstronomy/ImSim/image_linear_solve.py
@@ -969,7 +969,7 @@ def _image_linear_solve_interferometry_natwt_solving(self, A, d):
         for i in range(num_of_light):
             b[i] = np.sum(A[i] * (d))
 
-        param_amps = np.linalg.lstsq(M, b)[0]
+        param_amps = np.linalg.lstsq(M, b, rcond=None)[0]
 
         clean_temp = np.zeros((num_of_image_pixel))
         dirty_temp = np.zeros((num_of_image_pixel))

diff --git a/test/test_ImSim/test_image_linear_solve_with_interferometric_changes.py b/test/test_ImSim/test_image_linear_solve_with_interferometric_changes.py
@@ -158,7 +158,7 @@ def test_image_linear_solve_with_primary_beam_and_interferometry_psf():
     b[0] = np.sum(A0 * sim_data)
     b[1] = np.sum(A1 * sim_data)
 
-    amps0 = np.linalg.lstsq(M, b)[0]
+    amps0 = np.linalg.lstsq(M, b, rcond=None)[0]
     clean_model = amps0[0] * A0 + amps0[1] * A1
     dirty_model = amps0[0] * A0c + amps0[1] * A1c