Update firedrake/preconditioners/fdm.py

Co-authored-by: Connor Ward <[email protected]>
firedrakeproject · Feb 14, 2024 · 8099286 · 8099286
1 parent 29b478c
commit 8099286
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/firedrake/preconditioners/fdm.py b/firedrake/preconditioners/fdm.py
@@ -712,7 +712,7 @@ def make_args(self, *mats):
     def kernel(self, mat_type="aij", on_diag=False, addv=None):
         if addv is None:
             addv = PETSc.InsertMode.INSERT
-        indices = ("rindices", "cindices")[:2-on_diag]
+        indices = ("rindices",) if on_diag else ("rindices", "cindices")
         code = ""
         if "MatSetValuesArray" in self.code:
             code = """