lint

Author: leo-collins

firedrake/assemble.py

@@ -1585,8 +1585,8 @@ def allocate(self):
             appctx=self._appctx
         )
         return matrix.ImplicitMatrix(
-            self._form, ctx, self._bcs, 
-            fc_params=self._form_compiler_params, 
+            self._form, ctx, self._bcs,
+            fc_params=self._form_compiler_params,
             options_prefix=self._options_prefix
         )
 

firedrake/matrix.py

@@ -1,12 +1,12 @@
 from __future__ import annotations
-from typing import Any, Iterable, Literal, TYPE_CHECKING
+from typing import Any, Iterable, TYPE_CHECKING
 import itertools
 
 if TYPE_CHECKING:
     from firedrake.bcs import BCBase
     from firedrake.matrix_free.operators import ImplicitMatrixContext
     from firedrake.slate.slate import TensorBase
-    
+
 import ufl
 from ufl.argument import BaseArgument
 from pyop2 import op2
@@ -15,7 +15,6 @@
 from firedrake.petsc import PETSc
 
 
-
 class DummyOP2Mat:
     """A hashable implementation of M.handle"""
     def __init__(self, handle):
@@ -25,11 +24,11 @@ def __init__(self, handle):
 class MatrixBase(ufl.Matrix):
 
     def __init__(
-            self, 
-            a: ufl.BaseForm | TensorBase | tuple[BaseArgument, BaseArgument],
-            bcs: Iterable[BCBase] | None = None,
-            fc_params: dict[str, Any] | None = None,
-        ):
+        self,
+        a: ufl.BaseForm | TensorBase | tuple[BaseArgument, BaseArgument],
+        bcs: Iterable[BCBase] | None = None,
+        fc_params: dict[str, Any] | None = None,
+    ):
         """A representation of the linear operator associated with a bilinear form and bcs.
         Explicitly assembled matrices and matrix-free .matrix classes will derive from this.
 
@@ -145,13 +144,13 @@ def zero(self):
 class Matrix(MatrixBase):
 
     def __init__(
-            self,
-            a: ufl.BaseForm,
-            mat: op2.Mat | PETSc.Mat,
-            bcs: Iterable[BCBase] | None = None,
-            fc_params: dict[str, Any] | None = None,
-            options_prefix: str | None = None,
-        ):
+        self,
+        a: ufl.BaseForm,
+        mat: op2.Mat | PETSc.Mat,
+        bcs: Iterable[BCBase] | None = None,
+        fc_params: dict[str, Any] | None = None,
+        options_prefix: str | None = None,
+    ):
         """A representation of an assembled bilinear form.
 
         Parameters
@@ -189,15 +188,15 @@ def assemble(self):
 class ImplicitMatrix(MatrixBase):
 
     def __init__(
-            self,
-            a: ufl.BaseForm,
-            ctx: ImplicitMatrixContext,
-            bcs: Iterable[BCBase] | None = None,
-            fc_params: dict[str, Any] | None = None,
-            options_prefix: str | None = None,
-        ):
-        """A representation of the action of bilinear form operating without 
-        explicitly assembling the associated matrix. This class wraps the 
+        self,
+        a: ufl.BaseForm,
+        ctx: ImplicitMatrixContext,
+        bcs: Iterable[BCBase] | None = None,
+        fc_params: dict[str, Any] | None = None,
+        options_prefix: str | None = None,
+    ):
+        """A representation of the action of bilinear form operating without
+        explicitly assembling the associated matrix. This class wraps the
         relevant information for Python PETSc matrix.
 
         Parameters
@@ -238,12 +237,12 @@ def assemble(self):
 class AssembledMatrix(MatrixBase):
 
     def __init__(
-            self,
-            args: tuple[BaseArgument, BaseArgument],
-            petscmat: PETSc.Mat,
-            bcs: Iterable[BCBase] | None = None,
-            options_prefix: str | None = None,
-        ):
+        self,
+        args: tuple[BaseArgument, BaseArgument],
+        petscmat: PETSc.Mat,
+        bcs: Iterable[BCBase] | None = None,
+        options_prefix: str | None = None,
+    ):
         """A representation of a matrix that doesn't require knowing the underlying form.
 
         Parameters

firedrake/matrix_free/operators.py

@@ -69,16 +69,15 @@ class ImplicitMatrixContext(object):
     # (0,0) block of a 1x1 block matrix is on the diagonal).
     on_diag = True
 
-
     @PETSc.Log.EventDecorator()
     def __init__(
-        self, 
-        a: ufl.BaseForm, 
+        self,
+        a: ufl.BaseForm,
         row_bcs: Iterable[DirichletBC] | None = None,
         col_bcs: Iterable[DirichletBC] | None = None,
-        fc_params : dict[str, Any] | None = None, 
+        fc_params: dict[str, Any] | None = None,
         appctx: dict[str, Any] | None = None
-        ):
+    ):
         """This class gives the Python context for a PETSc Python matrix.
 
         Parameters
@@ -97,7 +96,7 @@ def __init__(
             A dictionary of parameters to pass on to the form compiler.
             By default None.
         appctx
-            Any extra user-supplied context, available to preconditioners 
+            Any extra user-supplied context, available to preconditioners
             and the like. By default None.
         """
         from firedrake.assemble import get_assembler