Skip to content

Commit

Permalink
Ruff formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
jorgensd committed May 3, 2024
1 parent f47b2a4 commit 55d8116
Show file tree
Hide file tree
Showing 4 changed files with 13 additions and 9 deletions.
5 changes: 3 additions & 2 deletions python/benchmarks/bench_contact_3D.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
7 changes: 4 additions & 3 deletions python/benchmarks/bench_elasticity.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down Expand Up @@ -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)
Expand Down
5 changes: 3 additions & 2 deletions python/benchmarks/bench_elasticity_edge.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
5 changes: 3 additions & 2 deletions python/benchmarks/ref_elasticity.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 55d8116

Please sign in to comment.