Skip to content

Commit 4b18d2a

Browse files
Re-run example notebooks
1 parent b8c2872 commit 4b18d2a

File tree

2 files changed

+444
-162
lines changed

2 files changed

+444
-162
lines changed

notebooks/Making a Custom Statespace Model.ipynb

Lines changed: 424 additions & 154 deletions
Large diffs are not rendered by default.

pymc_extras/statespace/core/statespace.py

Lines changed: 20 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -275,10 +275,27 @@ def __init__(
275275
self.requirement_table.add_column("Constraints", justify="left")
276276
self.requirement_table.add_column("Dimensions", justify="right")
277277

278-
self._populate_prior_requirements()
279-
self._populate_data_requirements()
278+
has_prior_info = False
279+
has_data_info = False
280+
try:
281+
self.param_info
282+
has_prior_info = True
283+
except NotImplementedError:
284+
pass
285+
286+
try:
287+
self.data_info
288+
has_data_info = True
289+
except NotImplementedError:
290+
pass
291+
292+
if has_prior_info:
293+
self._populate_prior_requirements()
280294

281-
if verbose:
295+
if has_data_info:
296+
self._populate_data_requirements()
297+
298+
if verbose and (has_prior_info or has_data_info):
282299
console = Console()
283300
console.print(self.requirement_table)
284301

@@ -296,11 +313,6 @@ def _populate_data_requirements(self) -> None:
296313
"""
297314
Add requirements about the data needed for the model, including their names, shapes, and named dimensions.
298315
"""
299-
try:
300-
self.data_info
301-
except NotImplementedError:
302-
return
303-
304316
self.requirement_table.add_section()
305317

306318
for data, info in self.data_info.items():

0 commit comments

Comments
 (0)