..

dweindl · Apr 16, 2024 · 4c0843f · 4c0843f
1 parent 3ef331a
commit 4c0843f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/python/sdist/amici/__init__.py b/python/sdist/amici/__init__.py
@@ -134,8 +134,6 @@ def get_model(self) -> amici.Model:
             """Create a model instance."""
             ...
 
-    AmiciModel = Union[amici.Model, amici.ModelPtr]
-
 
 class add_path:
     """Context manager for temporarily changing PYTHONPATH"""

diff --git a/swig/amici.i b/swig/amici.i
@@ -355,7 +355,7 @@ if sys.platform == 'win32' and (dll_dirs := os.environ.get('AMICI_DLL_DIRS')):
 // import additional types for typehints
 // also import np for use in __repr__ functions
 %pythonbegin %{
-from typing import TYPE_CHECKING, Iterable
+from typing import TYPE_CHECKING, Iterable, Union
 from collections.abc import Sequence
 import numpy as np
 if TYPE_CHECKING:
@@ -372,11 +372,11 @@ __all__ = [
     and x not in {"np", "sys", "os", "numpy", "IntEnum", "enum", "pi", "TYPE_CHECKING", "Iterable", "Sequence"}
 ]
 
-AmiciModel = Model | ModelPtr
-AmiciSolver = Solver | SolverPtr
-AmiciExpData = ExpData | ExpDataPtr
-AmiciReturnData = ReturnData | ReturnDataPtr
-AmiciExpDataVector = ExpDataPtrVector | Sequence[AmiciExpData]
+AmiciModel = Union['Model', 'ModelPtr']
+AmiciSolver = Union['Solver', 'SolverPtr']
+AmiciExpData = Union['ExpData', 'ExpDataPtr']
+AmiciReturnData = Union['ReturnData', 'ReturnDataPtr']
+AmiciExpDataVector = Union['ExpDataPtrVector', Sequence['AmiciExpData']]
 
 def _get_ptr(
     obj: AmiciModel | AmiciExpData | AmiciSolver | AmiciReturnData,