From c898f30a5b0cc56405b4e02137c7498ef9fd5776 Mon Sep 17 00:00:00 2001
From: Daniel Weindl <daniel.weindl@helmholtz-munich.de>
Date: Wed, 20 Dec 2023 14:27:05 +0100
Subject: [PATCH] imports, all

---
 python/sdist/amici/petab/__init__.py    | 16 ++++++++++++++++
 python/sdist/amici/petab/simulations.py | 13 +++++++++++++
 2 files changed, 29 insertions(+)

diff --git a/python/sdist/amici/petab/__init__.py b/python/sdist/amici/petab/__init__.py
index 920c0bce32..6d2201ce3b 100644
--- a/python/sdist/amici/petab/__init__.py
+++ b/python/sdist/amici/petab/__init__.py
@@ -6,6 +6,14 @@
 
 from .petab_import import import_petab_problem
 from .simulations import (
+    EDATAS,
+    FIM,
+    LLH,
+    RDATAS,
+    RES,
+    S2LLH,
+    SLLH,
+    SRES,
     rdatas_to_measurement_df,
     rdatas_to_simulation_df,
     simulate_petab,
@@ -16,4 +24,12 @@
     "simulate_petab",
     "rdatas_to_simulation_df",
     "rdatas_to_measurement_df",
+    "LLH",
+    "SLLH",
+    "FIM",
+    "S2LLH",
+    "RES",
+    "SRES",
+    "RDATAS",
+    "EDATAS",
 ]
diff --git a/python/sdist/amici/petab/simulations.py b/python/sdist/amici/petab/simulations.py
index 867679ca4d..bc42ebcfd3 100644
--- a/python/sdist/amici/petab/simulations.py
+++ b/python/sdist/amici/petab/simulations.py
@@ -54,6 +54,19 @@
 EDATAS = "edatas"
 
 
+__all__ = [
+    "simulate_petab",
+    "LLH",
+    "SLLH",
+    "FIM",
+    "S2LLH",
+    "RES",
+    "SRES",
+    "RDATAS",
+    "EDATAS",
+]
+
+
 @log_execution_time("Simulating PEtab model", logger)
 def simulate_petab(
     petab_problem: petab.Problem,