Check k_exog is not None

jessegrabowski · jessegrabowski · commit 90c92cd853f6 · 2025-09-23T10:45:15.000-05:00
diff --git a/pymc_extras/statespace/models/VARMAX.py b/pymc_extras/statespace/models/VARMAX.py
@@ -254,7 +254,7 @@ def __init__(
             mode=mode,
         )
 
-        self._needs_exog_data = k_exog > 0
+        self._needs_exog_data = k_exog is not None and k_exog > 0
 
         # Save counts of the number of parameters in each category
         self.param_counts = {

Original file line number	Diff line number	Diff line change
`@@ -254,7 +254,7 @@ def __init__(`
`254`	`254`	`mode=mode,`
`255`	`255`	`)`
`256`	`256`
`257`		`- self._needs_exog_data = k_exog > 0`
	`257`	`+ self._needs_exog_data = k_exog is not None and k_exog > 0`
`258`	`258`
`259`	`259`	`# Save counts of the number of parameters in each category`
`260`	`260`	`self.param_counts = {`