Re-run example notebooks

pymc-devs · Jan 6, 2025 · 4a86cad · 4a86cad
1 parent 2e5f0a6
commit 4a86cad
Show file tree

Hide file tree

Showing 2 changed files with 444 additions and 162 deletions.
diff --git a/notebooks/Making a Custom Statespace Model.ipynb b/notebooks/Making a Custom Statespace Model.ipynb
diff --git a/pymc_extras/statespace/core/statespace.py b/pymc_extras/statespace/core/statespace.py
@@ -275,10 +275,27 @@ def __init__(
         self.requirement_table.add_column("Constraints", justify="left")
         self.requirement_table.add_column("Dimensions", justify="right")
 
-        self._populate_prior_requirements()
-        self._populate_data_requirements()
+        has_prior_info = False
+        has_data_info = False
+        try:
+            self.param_info
+            has_prior_info = True
+        except NotImplementedError:
+            pass
+
+        try:
+            self.data_info
+            has_data_info = True
+        except NotImplementedError:
+            pass
+
+        if has_prior_info:
+            self._populate_prior_requirements()
 
-        if verbose:
+        if has_data_info:
+            self._populate_data_requirements()
+
+        if verbose and (has_prior_info or has_data_info):
             console = Console()
             console.print(self.requirement_table)
 
@@ -296,11 +313,6 @@ def _populate_data_requirements(self) -> None:
         """
         Add requirements about the data needed for the model, including their names, shapes, and named dimensions.
         """
-        try:
-            self.data_info
-        except NotImplementedError:
-            return
-
         self.requirement_table.add_section()
 
         for data, info in self.data_info.items():