Popultate __all__ for scores and links

han-ol · han-ol · commit 95cd9508c6be · 2025-03-05T15:04:35.000+01:00
diff --git a/bayesflow/diagnostics/__init__.py b/bayesflow/diagnostics/__init__.py
@@ -2,6 +2,7 @@
 
 from .plots import (
     calibration_ecdf,
+    calibration_ecdf_from_quantiles,
     calibration_histogram,
     loss,
     mc_calibration,
@@ -10,6 +11,7 @@
     pairs_posterior,
     pairs_samples,
     recovery,
+    recovery_from_estimates,
     z_score_contraction,
 )
 
diff --git a/bayesflow/links/__init__.py b/bayesflow/links/__init__.py
@@ -1,3 +1,7 @@
 from .ordered import Ordered
 from .ordered_quantiles import OrderedQuantiles
 from .positive_semi_definite import PositiveSemiDefinite
+
+from ..utils._docs import _add_imports_to_all
+
+_add_imports_to_all(include_modules=[])
diff --git a/bayesflow/scores/__init__.py b/bayesflow/scores/__init__.py
@@ -7,3 +7,7 @@
     QuantileScore,
     MultivariateNormalScore,
 )
+
+from ..utils._docs import _add_imports_to_all
+
+_add_imports_to_all(include_modules=[])

Original file line number	Diff line number	Diff line change
`@@ -7,3 +7,7 @@`
`7`	`7`	`QuantileScore,`
`8`	`8`	`MultivariateNormalScore,`
`9`	`9`	`)`
	`10`	`+`
	`11`	`+from ..utils._docs import _add_imports_to_all`
	`12`	`+`
	`13`	`+_add_imports_to_all(include_modules=[])`