Update PETSc vector creation API (FEniCS/dolfinx#3659)

Author: jorgensd

python/demos/create_and_export_mesh.py

@@ -394,7 +394,7 @@ def gmsh_2D_stacked(celltype: str, theta: float, verbose: bool = False) -> Tuple
     gmsh.clear()
     gmsh.finalize()
     MPI.COMM_WORLD.barrier()
-    return mesh,  mesh_data.facet_tags
+    return mesh, mesh_data.facet_tags
 
 
 def mesh_2D_dolfin(celltype: str, theta: float = 0, outdir: Union[str, Path] = Path("meshes")):

python/demos/demo_periodic_gep.py

@@ -354,7 +354,7 @@ def pbc_slave_to_master_map(x):
         mpc.backsubstitution(eigvec_r[i])
         eigvec_i[i].x.scatter_forward()
         mpc.backsubstitution(eigvec_i[i])
-    print0(f"Computed eigenvalues:\n {np.around(eigval,decimals=2)}")
+    print0(f"Computed eigenvalues:\n {np.around(eigval, decimals=2)}")
 
     # Save all eigenvectors
     suffix = "".join([bc_type[0] for bc_type in boundary_condition])
@@ -378,7 +378,7 @@ def print_exact_eigenvalues(boundary_condition: List[str], N: int):
         ev_y = [(n * 2 * np.pi / L[1]) ** 2 for n in range(-N, N)]
     ev_ex = np.sort([r + q for r in ev_x for q in ev_y])
     ev_ex = ev_ex[0:N]
-    print0(f"Exact eigenvalues (repeated with multiplicity):\n {np.around(ev_ex,decimals=2)}")
+    print0(f"Exact eigenvalues (repeated with multiplicity):\n {np.around(ev_ex, decimals=2)}")
 
 
 # Dirichlet boundary condition on {x=0} and {x=1}

python/dolfinx_mpc/assemble_vector.py

@@ -12,11 +12,11 @@
 
 import dolfinx.cpp as _cpp
 import dolfinx.fem as _fem
-import dolfinx.la as _la
 import numpy
 import ufl
 from dolfinx import default_scalar_type
 from dolfinx.common import Timer
+from dolfinx.la.petsc import create_vector
 
 import dolfinx_mpc.cpp
 
@@ -74,7 +74,7 @@ def assemble_vector(
     """
 
     if b is None:
-        b = _la.create_petsc_vector(
+        b = create_vector(
             constraint.function_space.dofmap.index_map,
             constraint.function_space.dofmap.index_map_bs,
         )

python/dolfinx_mpc/numba/assemble_vector.py

@@ -13,11 +13,11 @@
 import dolfinx
 import dolfinx.cpp as _cpp
 import dolfinx.fem as _fem
-import dolfinx.la as _la
 import dolfinx.log as _log
 import numpy
 import numpy.typing as npt
 from dolfinx.common import Timer
+from dolfinx.la.petsc import create_vector
 
 import numba
 from dolfinx_mpc.multipointconstraint import MultiPointConstraint
@@ -68,7 +68,7 @@ def assemble_vector(form: _forms, constraint: MultiPointConstraint, b: Optional[
     # Get index map and ghost info
     if b is None:
         index_map = constraint.function_space.dofmap.index_map
-        vector = _la.create_petsc_vector(index_map, block_size)
+        vector = create_vector(index_map, block_size)
     else:
         vector = b
 

python/dolfinx_mpc/problem.py

@@ -14,7 +14,7 @@
 import ufl
 from dolfinx import cpp as _cpp
 from dolfinx import fem as _fem
-from dolfinx import la as _la
+from dolfinx.la.petsc import create_vector
 
 from .assemble_matrix import assemble_matrix, create_sparsity_pattern
 from .assemble_vector import apply_lifting, assemble_vector
@@ -107,9 +107,7 @@ def __init__(
         pattern.finalize()
         self._A = _cpp.la.petsc.create_matrix(self._mpc.function_space.mesh.comm, pattern)
 
-        self._b = _la.create_petsc_vector(
-            self._mpc.function_space.dofmap.index_map, self._mpc.function_space.dofmap.index_map_bs
-        )
+        self._b = create_vector(self._mpc.function_space.dofmap.index_map, self._mpc.function_space.dofmap.index_map_bs)
         self.bcs = [] if bcs is None else bcs
 
         self._solver = PETSc.KSP().create(self.u.function_space.mesh.comm)  # type: ignore

python/tests/test_nonlinear_assembly.py

@@ -11,11 +11,11 @@
 import basix
 import dolfinx
 import dolfinx.fem.petsc
-import dolfinx.la as _la
 import dolfinx.nls.petsc
 import numpy as np
 import pytest
 import ufl
+from dolfinx.la.petsc import create_vector
 
 import dolfinx_mpc
 
@@ -63,7 +63,7 @@ def __init__(
         # Create matrix and vector to be used for assembly of the non-linear
         # MPC problem
         self._A = dolfinx_mpc.cpp.mpc.create_matrix(problem.a._cpp_object, mpc._cpp_object)
-        self._b = _la.create_petsc_vector(mpc.function_space.dofmap.index_map, mpc.function_space.dofmap.index_map_bs)
+        self._b = create_vector(mpc.function_space.dofmap.index_map, mpc.function_space.dofmap.index_map_bs)
 
         self.setF(problem.F, self._b)
         self.setJ(problem.J, self._A)