diff --git a/python/dolfinx/io.py b/python/dolfinx/io.py index 4bd9081aa4b..2e264a19d23 100644 --- a/python/dolfinx/io.py +++ b/python/dolfinx/io.py @@ -13,9 +13,9 @@ import ufl from dolfinx import cpp as _cpp -from dolfinx.fem import Function from dolfinx.cpp.io import distribute_entity_data from dolfinx.cpp.io import perm_gmsh as cell_perm_gmsh +from dolfinx.fem import Function from dolfinx.mesh import GhostMode, Mesh __all__ = ["VTKFile", "XDMFFile", "cell_perm_gmsh", "distribute_entity_data"] diff --git a/python/dolfinx/plot.py b/python/dolfinx/plot.py index 45a888137e5..ca4604dfee7 100644 --- a/python/dolfinx/plot.py +++ b/python/dolfinx/plot.py @@ -11,8 +11,7 @@ import numpy as np from dolfinx import cpp as _cpp -from dolfinx import fem -from dolfinx import mesh +from dolfinx import fem, mesh from dolfinx.mesh import CellType # NOTE: Edge visualization of higher order elements are sketchy, see: