fixup

AMICI-dev · Dec 3, 2024 · 0b1102b · 0b1102b
1 parent 9465a6c
commit 0b1102b
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 6 deletions.
diff --git a/python/sdist/amici/__init__.py b/python/sdist/amici/__init__.py
@@ -141,8 +141,6 @@ def get_model(self) -> amici.Model:
             """Create a model instance."""
             ...
 
-        def get_jax_model(self) -> JAXModel: ...
-
     AmiciModel = Union[amici.Model, amici.ModelPtr]
 else:
     ModelModule = ModuleType

diff --git a/python/sdist/amici/__init__.template.py b/python/sdist/amici/__init__.template.py
@@ -2,13 +2,9 @@
 
 import sys
 from pathlib import Path
-from typing import TYPE_CHECKING
 import amici
 
 
-if TYPE_CHECKING:
-    pass
-
 # Ensure we are binary-compatible, see #556
 if "TPL_AMICI_VERSION" != amici.__version__:
     raise amici.AmiciVersionError(