Add explicit imports of mpi4py, petsc4py and slepc4py submodules in s…

…ources, tests and tutorials

rbnicsx/_backends/export.py

@@ -8,8 +8,8 @@
 import os
 import typing
 
-import mpi4py
-import petsc4py
+import mpi4py.MPI
+import petsc4py.PETSc
 
 from rbnicsx.io import on_rank_zero
 

rbnicsx/_backends/functions_list.py

@@ -10,8 +10,8 @@
 import abc
 import typing
 
-import mpi4py
-import petsc4py
+import mpi4py.MPI
+import petsc4py.PETSc
 
 Function = typing.TypeVar("Function")
 

rbnicsx/_backends/import_.py

@@ -8,8 +8,8 @@
 import os
 import typing
 
-import mpi4py
-import petsc4py
+import mpi4py.MPI
+import petsc4py.PETSc
 
 from rbnicsx.io import on_rank_zero
 

rbnicsx/_backends/online_tensors.py

@@ -11,9 +11,9 @@
 import types
 import typing
 
-import mpi4py
+import mpi4py.MPI
 import numpy as np
-import petsc4py
+import petsc4py.PETSc
 
 
 def create_online_vector(N: int) -> petsc4py.PETSc.Vec:

rbnicsx/_backends/projection.py

@@ -7,7 +7,7 @@
 
 import typing
 
-import petsc4py
+import petsc4py.PETSc
 
 from rbnicsx._backends.functions_list import FunctionsList
 from rbnicsx._backends.online_tensors import BlockMatSubMatrixWrapper, BlockVecSubVectorWrapper

rbnicsx/_backends/proper_orthogonal_decomposition.py

@@ -8,8 +8,8 @@
 import typing
 
 import numpy as np
-import petsc4py
-import slepc4py
+import petsc4py.PETSc
+import slepc4py.SLEPc
 
 from rbnicsx._backends.functions_list import Function, FunctionsList
 from rbnicsx._backends.online_tensors import create_online_matrix, create_online_vector

rbnicsx/_backends/tensors_array.py

@@ -12,9 +12,9 @@
 import os
 import typing
 
-import mpi4py
+import mpi4py.MPI
 import numpy as np
-import petsc4py
+import petsc4py.PETSc
 
 from rbnicsx.io import on_rank_zero
 

rbnicsx/_backends/tensors_list.py

@@ -11,8 +11,8 @@
 import os
 import typing
 
-import mpi4py
-import petsc4py
+import mpi4py.MPI
+import petsc4py.PETSc
 
 from rbnicsx.io import on_rank_zero
 

rbnicsx/_cpp/__init__.py

@@ -7,7 +7,7 @@
 
 import os
 
-import mpi4py
+import mpi4py.MPI
 
 from rbnicsx._cpp.compile_code import compile_code
 from rbnicsx._cpp.compile_package import compile_package
@@ -32,8 +32,3 @@
     os.path.dirname(os.path.abspath(__file__)),
     *sources
 )
-
-# Clean up imports
-del mpi4py
-
-del os

rbnicsx/_cpp/compile_code.py

@@ -13,6 +13,7 @@
 
 import cppimport
 import mpi4py
+import mpi4py.MPI
 import numpy
 import petsc4py
 import pybind11

rbnicsx/_cpp/compile_package.py

@@ -10,7 +10,7 @@
 import types
 import typing
 
-import mpi4py
+import mpi4py.MPI
 
 from rbnicsx._cpp.compile_code import compile_code
 

rbnicsx/backends/export.py

@@ -11,7 +11,7 @@
 import dolfinx.fem
 import dolfinx.io
 import numpy as np
-import petsc4py
+import petsc4py.PETSc
 
 from rbnicsx._backends.export import (
     export_matrices as export_matrices_super, export_matrix as export_matrix_super,

rbnicsx/backends/functions_list.py

@@ -9,7 +9,7 @@
 
 import dolfinx.fem
 import numpy as np
-import petsc4py
+import petsc4py.PETSc
 
 from rbnicsx._backends.functions_list import FunctionsList as FunctionsListBase
 from rbnicsx.backends.export import export_functions

rbnicsx/backends/gram_schmidt.py

@@ -9,7 +9,7 @@
 
 import dolfinx.fem
 import numpy as np
-import petsc4py
+import petsc4py.PETSc
 
 from rbnicsx.backends.functions_list import FunctionsList
 

rbnicsx/backends/import_.py

@@ -9,8 +9,8 @@
 import typing
 
 import dolfinx.fem
-import mpi4py
-import petsc4py
+import mpi4py.MPI
+import petsc4py.PETSc
 
 from rbnicsx._backends.import_ import (
     import_matrices as import_matrices_super, import_matrix as import_matrix_super,

rbnicsx/backends/mesh_motion.py

@@ -13,7 +13,7 @@
 import dolfinx.fem
 import dolfinx.mesh
 import numpy as np
-import petsc4py
+import petsc4py.PETSc
 
 
 class MeshMotion(object):

rbnicsx/backends/projection.py

@@ -9,9 +9,9 @@
 import typing
 
 import dolfinx.fem
-import mpi4py
+import mpi4py.MPI
 import numpy as np
-import petsc4py
+import petsc4py.PETSc
 import ufl
 
 from rbnicsx._backends.online_tensors import (

rbnicsx/backends/proper_orthogonal_decomposition.py

@@ -10,7 +10,7 @@
 
 import dolfinx.fem
 import numpy as np
-import petsc4py
+import petsc4py.PETSc
 
 from rbnicsx._backends.proper_orthogonal_decomposition import (
     proper_orthogonal_decomposition_functions as proper_orthogonal_decomposition_functions_super,

rbnicsx/backends/symbolic_parameters.py

@@ -9,7 +9,7 @@
 
 import dolfinx.mesh
 import numpy as np
-import petsc4py
+import petsc4py.PETSc
 import ufl4rom.utils
 
 

rbnicsx/backends/tensors_array.py

@@ -10,7 +10,7 @@
 import typing
 
 import dolfinx.fem
-import mpi4py
+import mpi4py.MPI
 import numpy as np
 
 from rbnicsx._backends.tensors_array import TensorsArray as TensorsArrayBase

rbnicsx/backends/tensors_list.py

@@ -8,7 +8,7 @@
 from __future__ import annotations
 
 import dolfinx.fem
-import mpi4py
+import mpi4py.MPI
 
 from rbnicsx._backends.tensors_list import TensorsList as TensorsListBase
 from rbnicsx.backends.export import export_matrices, export_vectors

rbnicsx/io/on_rank_zero.py

@@ -8,8 +8,8 @@
 import sys
 import typing
 
-import mpi4py
-import petsc4py
+import mpi4py.MPI
+import petsc4py.PETSc
 
 
 def on_rank_zero(

rbnicsx/io/timer.py

@@ -11,8 +11,8 @@
 import types
 import typing
 
-import mpi4py
-import petsc4py
+import mpi4py.MPI
+import petsc4py.PETSc
 
 
 class Timer(object):

rbnicsx/online/export.py

@@ -7,8 +7,8 @@
 
 import typing
 
-import mpi4py
-import petsc4py
+import mpi4py.MPI
+import petsc4py.PETSc
 
 from rbnicsx._backends.export import (
     export_matrices as export_matrices_super, export_matrix as export_matrix_super,

rbnicsx/online/functions_list.py

@@ -9,8 +9,8 @@
 
 import typing
 
-import mpi4py
-import petsc4py
+import mpi4py.MPI
+import petsc4py.PETSc
 
 from rbnicsx._backends.functions_list import FunctionsList as FunctionsListBase
 from rbnicsx._backends.online_tensors import create_online_vector as create_vector

rbnicsx/online/gram_schmidt.py

@@ -8,7 +8,7 @@
 import typing
 
 import numpy as np
-import petsc4py
+import petsc4py.PETSc
 
 from rbnicsx.online.functions_list import FunctionsList
 from rbnicsx.online.projection import matrix_action

rbnicsx/online/import_.py

@@ -7,8 +7,8 @@
 
 import typing
 
-import mpi4py
-import petsc4py
+import mpi4py.MPI
+import petsc4py.PETSc
 
 from rbnicsx._backends.import_ import (
     import_matrices as import_matrices_super, import_matrix as import_matrix_super,

rbnicsx/online/projection.py

@@ -8,7 +8,7 @@
 import typing
 
 import multipledispatch
-import petsc4py
+import petsc4py.PETSc
 
 from rbnicsx._backends.online_tensors import (
     BlockMatSubMatrixCopier, BlockVecSubVectorCopier, create_online_matrix as create_matrix,

rbnicsx/online/proper_orthogonal_decomposition.py

@@ -9,7 +9,7 @@
 import typing
 
 import numpy as np
-import petsc4py
+import petsc4py.PETSc
 
 from rbnicsx._backends.proper_orthogonal_decomposition import (
     proper_orthogonal_decomposition_functions as proper_orthogonal_decomposition_functions_super,

rbnicsx/online/tensors_array.py

@@ -9,9 +9,9 @@
 
 import typing
 
-import mpi4py
+import mpi4py.MPI
 import numpy as np
-import petsc4py
+import petsc4py.PETSc
 
 from rbnicsx._backends.online_tensors import create_online_vector as create_vector
 from rbnicsx._backends.tensors_array import TensorsArray as TensorsArrayBase

rbnicsx/online/tensors_list.py

@@ -9,7 +9,7 @@
 
 import typing
 
-import mpi4py
+import mpi4py.MPI
 
 from rbnicsx._backends.tensors_list import TensorsList as TensorsListBase
 from rbnicsx.online.export import export_matrices, export_matrices_block, export_vectors, export_vectors_block

tests/unit/_cpp/_backends/test_frobenius_inner_product.py

@@ -5,8 +5,8 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
 """Tests for Frobenius inner product implementation in C++."""
 
-import mpi4py
-import petsc4py
+import mpi4py.MPI
+import petsc4py.PETSc
 
 import rbnicsx._cpp
 

tests/unit/_cpp/test_compile_cpp.py

@@ -7,7 +7,7 @@
 
 import os
 
-import mpi4py
+import mpi4py.MPI
 import nbvalx.tempfile
 import pytest
 

tests/unit/backends/test_backends_export_import.py

@@ -9,10 +9,10 @@
 
 import dolfinx.fem
 import dolfinx.mesh
-import mpi4py
+import mpi4py.MPI
 import nbvalx.tempfile
 import numpy as np
-import petsc4py
+import petsc4py.PETSc
 import pytest
 import ufl
 

tests/unit/backends/test_backends_functions_list.py

@@ -7,7 +7,7 @@
 
 import dolfinx.fem
 import dolfinx.mesh
-import mpi4py
+import mpi4py.MPI
 import nbvalx.tempfile
 import numpy as np
 import pytest

tests/unit/backends/test_backends_gram_schmidt.py

@@ -9,7 +9,7 @@
 
 import dolfinx.fem
 import dolfinx.mesh
-import mpi4py
+import mpi4py.MPI
 import numpy as np
 import pytest
 import ufl

tests/unit/backends/test_backends_mesh_motion.py

@@ -10,7 +10,7 @@
 import _pytest.fixtures
 import dolfinx.fem
 import dolfinx.mesh
-import mpi4py
+import mpi4py.MPI
 import numpy as np
 import pytest
 

tests/unit/backends/test_backends_projection.py

@@ -9,7 +9,7 @@
 
 import dolfinx.fem
 import dolfinx.mesh
-import mpi4py
+import mpi4py.MPI
 import numpy as np
 import pytest
 import ufl