From 55d811660a1a5f7ab538762911f6350e43647244 Mon Sep 17 00:00:00 2001 From: "Jorgen S. Dokken" Date: Fri, 3 May 2024 12:25:46 +0000 Subject: [PATCH] Ruff formatting --- python/benchmarks/bench_contact_3D.py | 5 +++-- python/benchmarks/bench_elasticity.py | 7 ++++--- python/benchmarks/bench_elasticity_edge.py | 5 +++-- python/benchmarks/ref_elasticity.py | 5 +++-- 4 files changed, 13 insertions(+), 9 deletions(-) diff --git a/python/benchmarks/bench_contact_3D.py b/python/benchmarks/bench_contact_3D.py index fbffa31..d9e36e9 100644 --- a/python/benchmarks/bench_contact_3D.py +++ b/python/benchmarks/bench_contact_3D.py @@ -211,8 +211,9 @@ def demo_stacked_cubes(theta, ct, noslip, num_refinements, N0, timings=False): mesh.name = f"mesh_{celltype}_{theta:.2f}{type_ext:s}" # Create functionspaces - el = basix.ufl.element("Lagrange", mesh.topology.cell_name(), 1, shape=(mesh.geometry.dim,), - dtype=default_real_type) + el = basix.ufl.element( + "Lagrange", mesh.topology.cell_name(), 1, shape=(mesh.geometry.dim,), dtype=default_real_type + ) V = functionspace(mesh, el) # Define boundary conditions diff --git a/python/benchmarks/bench_elasticity.py b/python/benchmarks/bench_elasticity.py index f998b85..aedbc39 100644 --- a/python/benchmarks/bench_elasticity.py +++ b/python/benchmarks/bench_elasticity.py @@ -16,7 +16,7 @@ import basix.ufl import h5py import numpy as np -from dolfinx import default_real_type,default_scalar_type +from dolfinx import default_real_type, default_scalar_type from dolfinx.common import Timer, TimingType, list_timings from dolfinx.fem import ( Constant, @@ -49,8 +49,9 @@ def bench_elasticity_one( mesh = refine(mesh, redistribute=True) fdim = mesh.topology.dim - 1 - el = basix.ufl.element("Lagrange", mesh.topology.cell_name(), 1, - shape=(mesh.geometry.dim,), dtype=default_real_type) + el = basix.ufl.element( + "Lagrange", mesh.topology.cell_name(), 1, shape=(mesh.geometry.dim,), dtype=default_real_type + ) V = functionspace(mesh, el) # Generate Dirichlet BC on lower boundary (Fixed) diff --git a/python/benchmarks/bench_elasticity_edge.py b/python/benchmarks/bench_elasticity_edge.py index e57f14f..237f689 100644 --- a/python/benchmarks/bench_elasticity_edge.py +++ b/python/benchmarks/bench_elasticity_edge.py @@ -62,8 +62,9 @@ def bench_elasticity_edge( ct = CellType.tetrahedron if tetra else CellType.hexahedron mesh = create_unit_cube(MPI.COMM_WORLD, N, N, N, ct) - el = basix.ufl.element("Lagrange", mesh.topology.cell_name(), int(degree), shape=(mesh.geometry.dim,), - dtype=default_real_type) + el = basix.ufl.element( + "Lagrange", mesh.topology.cell_name(), int(degree), shape=(mesh.geometry.dim,), dtype=default_real_type + ) V = functionspace(mesh, el) # Generate Dirichlet BC (Fixed) diff --git a/python/benchmarks/ref_elasticity.py b/python/benchmarks/ref_elasticity.py index 615cf25..c1eee62 100644 --- a/python/benchmarks/ref_elasticity.py +++ b/python/benchmarks/ref_elasticity.py @@ -73,8 +73,9 @@ def ref_elasticity( # set_log_level(LogLevel.ERROR) N = degree * N fdim = mesh.topology.dim - 1 - el = basix.ufl.element("Lagrange", mesh.topology.cell_name(), 1, shape=(mesh.geometry.dim,), - dtype=default_real_type) + el = basix.ufl.element( + "Lagrange", mesh.topology.cell_name(), 1, shape=(mesh.geometry.dim,), dtype=default_real_type + ) V = functionspace(mesh, el) # Generate Dirichlet BC on lower boundary (Fixed)