Fix quantile level serialization and add save/load to notebook

Author: han-ol

bayesflow/approximators/point_approximator.py

@@ -24,9 +24,6 @@ def estimate(
         split: bool = False,
         **kwargs,
     ) -> dict[str, dict[str, np.ndarray]]:
-        if not self.built:
-            raise AssertionError("PointApproximator needs to be built before predicting with it.")
-
         conditions = self._prepare_conditions(conditions, **kwargs)
         estimates = self._estimate(**conditions, **kwargs)
         estimates = self._apply_inverse_adapter_to_estimates(estimates, **kwargs)
@@ -48,9 +45,6 @@ def sample(
         split: bool = False,
         **kwargs,
     ) -> dict[str, np.ndarray]:
-        if not self.built:
-            raise AssertionError("This model needs to be built before using it for sampling.")
-
         conditions = self._prepare_conditions(conditions, **kwargs)
         samples = self._sample(num_samples, **conditions, **kwargs)
         samples = self._apply_inverse_adapter_to_samples(samples, **kwargs)

bayesflow/scores/quantile_score.py

@@ -23,6 +23,8 @@ def __init__(self, q: Sequence[float] = None, links=None, **kwargs):
             q = [0.1, 0.5, 0.9]
             logging.info(f"QuantileScore was not provided with argument `q`. Using the default quantile levels: {q}.")
 
+        # force a conversion to list for proper serialization
+        q = list(q)
         self.q = q
         self._q = keras.ops.convert_to_tensor(q, dtype="float32")
         self.links = links or {"value": OrderedQuantiles(q=q)}

examples/Lotka_Volterra_point_estimation_and_expert_stats.ipynb

@@ -1,4 +1,5 @@
 import pytest
+import numpy as np
 
 
 @pytest.fixture()
@@ -72,5 +73,5 @@ def quantile_point_inference_network():
     from bayesflow.scores import QuantileScore
 
     return PointInferenceNetwork(
-        scores=dict(quantiles=QuantileScore(q=[0.1, 0.4, 0.5, 0.7], subnets=dict(value="mlp"))),
+        scores=dict(quantiles=QuantileScore(q=np.array([0.1, 0.4, 0.5, 0.7]), subnets=dict(value="mlp"))),
     )