fix petab version

ICB-DCM · Oct 24, 2024 · 0302a2d · 0302a2d
1 parent ce65b28
commit 0302a2d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/pyabc/petab/__init__.py b/pyabc/petab/__init__.py
@@ -4,5 +4,16 @@
 
 Problem definitions in the PEtab format (https://petab.rtfd.io).
 """
+import warnings
 
 from .amici import AmiciPetabImporter
+
+try:
+    import petab
+except ImportError:
+    warnings.warn(
+        "PEtab import requires an installation of petab "
+        "(https://github.com/PEtab-dev/PEtab). "
+        "Install via `pip3 install petab`.",
+        stacklevel=1,
+    )
diff --git a/test/petab/test_petab.py b/test/petab/test_petab.py
@@ -18,9 +18,7 @@
 
 
 @pytest.fixture(
-    params=itertools.product(
-        [petab.C.LIN, petab.C.LOG, petab.C.LOG10], [*petab.C.PRIOR_TYPES, None]
-    )
+    params=itertools.product([C.LIN, C.LOG, C.LOG10], [*C.PRIOR_TYPES, None])
 )
 def prior_specs(request):
     """A one-line parameter df for a given prior type."""