Lighter switch to ruff

.circleci/config.yml

@@ -11,13 +11,12 @@ install-python-components: &install-python-components
     pip3 install git+https://github.com/FEniCS/ufl.git
     pip3 install git+https://github.com/FEniCS/ffcx.git
 
-flake8-python-code: &flake8-python-code
+ruff-isort-python-code: &ruff-isort-python-code
   name: Flake8 checks on Python code
   command: |
     cd python/
-    python3 -m flake8 dolfinx
-    python3 -m flake8 demo
-    python3 -m flake8 test
+    ruff check .
+    isort --check .
 
 configure-cpp: &configure-cpp
   name: Configure (C++)
@@ -101,7 +100,7 @@ jobs:
     steps:
       - checkout
       - run: *install-python-components
-      - run: *flake8-python-code
+      - run: *ruff-isort-python-code
       - run: *configure-cpp
       - run: *build-install-cpp
 

.github/workflows/ccpp.yml

@@ -60,24 +60,15 @@ jobs:
           python3 -m pip install git+https://github.com/FEniCS/basix.git@${{ github.event.inputs.basix_ref }}
           python3 -m pip install git+https://github.com/FEniCS/ffcx.git@${{ github.event.inputs.ffcx_ref }}
 
-      - name: Flake8 checks
-        run: |
-          cd cpp/
-          python3 -m flake8 cmake
-          python3 -m flake8 demo
-          python3 -m flake8 test
-          cd ..
-          cd python/
-          python3 -m flake8 dolfinx
-          python3 -m flake8 demo
-          python3 -m flake8 test
-      - name: isort checks (non-blocking)
-        continue-on-error: true
+      - name: ruff and isort C++ .py file checks
         run: |
           cd cpp/
+          ruff check .
           python3 -m isort --check .
-          cd ..
+      - name: ruff and isort Python interface checks
+        run: |
           cd python/
+          ruff check .
           python3 -m isort --check .
       - name: mypy checks
         run: |

cpp/ruff.toml

@@ -0,0 +1 @@
+extend = "../python/pyproject.toml"

python/demo/demo_lagrange_variants.py

@@ -18,6 +18,7 @@
 # We begin this demo by importing the required modules.
 
 import typing
+
 from mpi4py import MPI
 
 # +
@@ -26,7 +27,7 @@
 
 import basix
 import basix.ufl
-import ufl   # type: ignore
+import ufl  # type: ignore
 from dolfinx import default_scalar_type, fem, mesh
 from dolfinx.fem.petsc import LinearProblem
 from ufl import ds, dx, grad, inner

python/demo/demo_static-condensation.py

@@ -42,7 +42,8 @@
 from dolfinx.io import XDMFFile
 from dolfinx.jit import ffcx_jit
 from dolfinx.mesh import locate_entities_boundary, meshtags
-from ffcx.codegeneration.utils import numba_ufcx_kernel_signature as ufcx_signature
+from ffcx.codegeneration.utils import \
+    numba_ufcx_kernel_signature as ufcx_signature
 
 if default_real_type == np.float32:
     print("float32 not yet supported for this demo.")

python/demo/demo_tnt-elements.py

@@ -23,7 +23,7 @@
 from mpi4py import MPI
 
 # +
-import matplotlib
+import matplotlib as mpl
 import matplotlib.pylab as plt
 import numpy as np
 
@@ -34,7 +34,7 @@
 from ufl import (SpatialCoordinate, TestFunction, TrialFunction, cos, div, dx,
                  grad, inner, sin)
 
-matplotlib.use('agg')
+mpl.use('agg')
 # -
 
 # ## Defining a degree 1 TNT element

python/dolfinx/__init__.py

@@ -20,7 +20,8 @@
 
 from dolfinx import common
 from dolfinx import cpp as _cpp
-from dolfinx import fem, geometry, graph, io, jit, la, log, mesh, nls, plot, utils
+from dolfinx import (fem, geometry, graph, io, jit, la, log, mesh, nls, plot,
+                     utils)
 # Initialise logging
 from dolfinx.common import (TimingType, git_commit_hash, has_debug, has_kahip,
                             has_parmetis, list_timings, timing)

python/dolfinx/fem/assemble.py

@@ -213,7 +213,7 @@ def _assemble_vector_array(b: np.ndarray, L: Form, constants=None, coeffs=None):
 
 
 @functools.singledispatch
-def assemble_matrix(a: typing.Any, bcs: typing.Optional[typing.List[DirichletBC]] = None,
+def assemble_matrix(a: typing.Any, bcs: typing.Optional[list[DirichletBC]] = None,
                     diagonal: float = 1.0, constants=None, coeffs=None,
                     block_mode: typing.Optional[la.BlockMode] = None):
     """Assemble bilinear form into a matrix.
@@ -246,7 +246,7 @@ def assemble_matrix(a: typing.Any, bcs: typing.Optional[typing.List[DirichletBC]
 
 
 @assemble_matrix.register
-def _assemble_matrix_csr(A: la.MatrixCSR, a: Form, bcs: typing.Optional[typing.List[DirichletBC]] = None,
+def _assemble_matrix_csr(A: la.MatrixCSR, a: Form, bcs: typing.Optional[list[DirichletBC]] = None,
                          diagonal: float = 1.0, constants=None, coeffs=None) -> la.MatrixCSR:
     """Assemble bilinear form into a matrix.
 
@@ -284,9 +284,9 @@ def _assemble_matrix_csr(A: la.MatrixCSR, a: Form, bcs: typing.Optional[typing.L
 # -- Modifiers for Dirichlet conditions ---------------------------------------
 
 
-def apply_lifting(b: np.ndarray, a: typing.List[Form],
-                  bcs: typing.List[typing.List[DirichletBC]],
-                  x0: typing.Optional[typing.List[np.ndarray]] = None,
+def apply_lifting(b: np.ndarray, a: list[Form],
+                  bcs: list[list[DirichletBC]],
+                  x0: typing.Optional[list[np.ndarray]] = None,
                   scale: float = 1.0, constants=None, coeffs=None) -> None:
     """Modify RHS vector b for lifting of Dirichlet boundary conditions.
 
@@ -316,7 +316,7 @@ def apply_lifting(b: np.ndarray, a: typing.List[Form],
     _cpp.fem.apply_lifting(b, _a, constants, coeffs, _bcs, x0, scale)
 
 
-def set_bc(b: np.ndarray, bcs: typing.List[DirichletBC],
+def set_bc(b: np.ndarray, bcs: list[DirichletBC],
            x0: typing.Optional[np.ndarray] = None, scale: float = 1.0) -> None:
     """Insert boundary condition values into vector.
 

python/dolfinx/fem/bcs.py

@@ -185,7 +185,7 @@ def dirichletbc(value: typing.Union[Function, Constant, np.ndarray],
 
 
 def bcs_by_block(spaces: typing.Iterable[typing.Union[dolfinx.fem.FunctionSpace, None]],
-                 bcs: typing.Iterable[DirichletBC]) -> typing.List[typing.List[DirichletBC]]:
+                 bcs: typing.Iterable[DirichletBC]) -> list[list[DirichletBC]]:
     """Arrange Dirichlet boundary conditions by the function space that
     they constrain.
 

python/dolfinx/fem/forms.py

@@ -62,7 +62,7 @@ def rank(self) -> int:
         return self._cpp_object.rank  # type: ignore
 
     @property
-    def function_spaces(self) -> typing.List[FunctionSpace]:
+    def function_spaces(self) -> list[FunctionSpace]:
         """Function spaces on which this form is defined"""
         return self._cpp_object.function_spaces  # type: ignore
 

python/dolfinx/fem/function.py

@@ -455,7 +455,7 @@ def sub(self, i: int) -> Function:
             returned `Function` should be stored by the caller to avoid
             repeated re-computation of the subspac.
         """
-        return Function(self._V.sub(i), self.x, name=f"{str(self)}_{i}")
+        return Function(self._V.sub(i), self.x, name=f"{self!s}_{i}")
 
     def split(self) -> tuple[Function, ...]:
         """Extract (any) sub-functions.
@@ -492,13 +492,13 @@ class ElementMetaData(typing.NamedTuple):
     """
     family: str
     degree: int
-    shape: typing.Optional[typing.Tuple[int, ...]] = None
+    shape: typing.Optional[tuple[int, ...]] = None
     symmetry: typing.Optional[bool] = None
 
 
 def functionspace(mesh: Mesh,
                   element: typing.Union[ufl.FiniteElementBase, ElementMetaData,
-                                        typing.Tuple[str, int, typing.Tuple, bool]],
+                                        tuple[str, int, tuple, bool]],
                   form_compiler_options: typing.Optional[dict[str, typing.Any]] = None,
                   jit_options: typing.Optional[dict[str, typing.Any]] = None) -> FunctionSpace:
     """Create a finite element function space.