satisfy CI pipeline

.github/workflows/ci.yml

@@ -56,4 +56,4 @@ jobs:
 
       - name: Test examples
         run: >-
-          for f in examples/*.py; do python "$f"; done
+          for f in examples/*.py; do echo "run $f" && python "$f"; done

examples/model.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-import equinox as eqx
 import jax
 import jax.numpy as jnp
 
@@ -85,8 +84,6 @@ def create_model():
 
 model = create_model()
 
-eqx.tree_pprint(model)
-
 init_values = model.parameter_values
 observation = jnp.array([37])
 asimov = model.evaluate().expectation()

examples/nll_profiling.py

@@ -5,8 +5,8 @@
 import equinox as eqx
 import jax
 import jax.numpy as jnp
-from examples.model import asimov, model, optimizer
 from jax.config import config
+from model import asimov, model, optimizer
 
 from dilax.likelihood import NLL
 from dilax.model import Model