Minor import fixes

Author: jorgensd

src/festim/hydrogen_transport_problem.py

@@ -17,12 +17,12 @@
 import festim.boundary_conditions
 import festim.problem
 from festim import (
-    SolubilityLaw,
     boundary_conditions,
     exports,
     k_B,
     problem,
 )
+from festim.material import SolubilityLaw
 from festim import (
     reaction as _reaction,
 )
@@ -39,7 +39,7 @@
     get_interpolation_points,
     is_it_time_to_export,
 )
-from festim.mesh import CoordinateSystem, Mesh
+from .mesh import CoordinateSystem, Mesh
 
 __all__ = [
     "HydrogenTransportProblem",

src/festim/mesh/__init__.py

@@ -1,5 +1,5 @@
-from .mesh import Mesh
+from .mesh import CoordinateSystem, Mesh
 from .mesh_1d import Mesh1D
 from .mesh_from_xdmf import MeshFromXDMF
 
-__all__ = ["Mesh", "Mesh1D", "MeshFromXDMF"]
+__all__ = ["CoordinateSystem", "Mesh", "Mesh1D", "MeshFromXDMF"]

src/festim/mesh/mesh.py

@@ -6,6 +6,7 @@
 from dolfinx.mesh import Mesh as dolfinx_Mesh
 from dolfinx.mesh import meshtags
 
+__all__= ["CoordinateSystem", "Mesh"]
 
 class CoordinateSystem(Enum):
     CARTESIAN = 10