add viz notebook to rtd

PEtab-dev · Nov 18, 2024 · 14adba0 · 14adba0
1 parent f3ef04e
commit 14adba0
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/doc/examples.rst b/doc/examples.rst
@@ -7,5 +7,6 @@ Various example notebooks.
    :maxdepth: 1
 
    examples/example_cli_famos.ipynb
+   examples/visualization.ipynb
    examples/workflow_cli.ipynb
    examples/workflow_python.ipynb
diff --git a/petab_select/plot.py b/petab_select/plot.py
@@ -20,6 +20,16 @@
 NORMAL_NODE_COLOR = "darkgrey"
 
 
+__all__ = [
+    "bar_criterion_vs_models",
+    "graph_history",
+    "graph_iteration_layers",
+    "line_selected",
+    "scatter_criterion_vs_n_estimated",
+    "upset",
+]
+
+
 def default_label_maker(model: Model) -> str:
     """Create a model label, for plotting."""
     return model.model_hash[:4]
@@ -86,10 +96,9 @@ def upset(
         The plot axes (see documentation from the `upsetplot` package).
     """
     # Get delta criterion values
-    values = np.array(
+    values = get_relative_criterion_values(
         [model.get_criterion(Criterion.AICC) for model in models]
     )
-    values -= values.min()
 
     # Sort by criterion value
     index = np.argsort(values)