fix LSMR cond cutoff

Author: patrick-kidger

.github/workflows/run_tests.yml

@@ -32,7 +32,7 @@ jobs:
 
       - name: Test with pytest
         run: |
-          pytest
+          python -m tests
 
       - name: Check that documentation can be built.
         run: |

tests/helpers.py

@@ -59,6 +59,8 @@ def _construct_matrix_impl(getkey, cond_cutoff, tags, size, dtype, i):
 def construct_matrix(getkey, solver, tags, num=1, *, size=3, dtype=jnp.float64):
     if isinstance(solver, lx.Normal):
         cond_cutoff = math.sqrt(1000)
+    elif isinstance(solver, lx.LSMR):
+        cond_cutoff = 10  # it's not doing super well for some reason
     else:
         cond_cutoff = 1000
     return tuple(