Clean up amici.sim.sundials exports

* Re-export `import_model_module`.
* Set `__all__`

The goal should be that after having compiled a model, for most applications, the only required import for loading and simulating that model is `amici.sim.sundials`.

Author: dweindl

python/sdist/amici/sim/sundials/__init__.py

@@ -11,7 +11,8 @@
 from types import ModuleType
 from typing import Protocol, runtime_checkable
 
-from ... import amici_path
+import amici
+from amici import amici_path, import_model_module
 
 #: boolean indicating if this is the full package with swig interface or
 #  the raw package without extension
@@ -22,6 +23,11 @@
 #: boolean indicating if amici was compiled with hdf5 support
 hdf5_enabled: bool = False
 
+__all__ = [
+    "import_model_module",
+    "ModelModule",
+]
+
 if has_clibs:
     # Import SWIG module and swig-dependent submodules
     #  if required and available
@@ -37,7 +43,7 @@
             message="builtin type .* has no __module__ attribute",
         )
         # The swig-generated Python module
-        from amici._installation import amici
+        from amici._installation import amici as amici_swig_py
 
     # TODO: selective import to avoid loading unneeded symbols
     from amici._installation.amici import *
@@ -47,6 +53,7 @@
     #  from .swig_wrappers
     hdf5_enabled = "read_solver_settings_from_hdf5" in dir()
     # These modules require the swig interface and other dependencies
+    from . import _pandas, _swig_wrappers
     from ._numpy import ExpDataView as ExpDataView
     from ._numpy import ReturnDataView as ReturnDataView
     from ._numpy import evaluate as evaluate
@@ -62,10 +69,24 @@ class ModelModule(Protocol):  # noqa: F811
 
         To enable static type checking."""
 
-        def get_model(self) -> amici.Model:
+        def get_model(self) -> amici_swig_py.Model:
             """Create a model instance."""
             ...
 
-    AmiciModel = amici.Model | amici.ModelPtr
+    AmiciModel = amici_swig_py.Model | amici_swig_py.ModelPtr
+
+    __all__ += [
+        *amici._installation.amici.__all__,
+        "ExpDataView",
+        "ReturnDataView",
+        "evaluate",
+        "ModelModule",
+        *_pandas.__all__,
+        *_swig_wrappers.__all__,
+        AmiciModel,
+    ]
+
+    # expose the swig module itself
+    amici = amici_swig_py
 else:
     ModelModule = ModuleType