Revert a few more changes

fonnesbeck · fonnesbeck · commit 58bc697c4466 · 2024-12-11T21:01:37.000-06:00
diff --git a/pymc_extras/statespace/core/statespace.py b/pymc_extras/statespace/core/statespace.py
@@ -54,9 +54,9 @@
 
 
 def _validate_filter_arg(filter_arg):
-    if filter_arg.lower() not in FILTER_FACTORY.keys():
+    if filter_arg.lower() not in FILTER_OUTPUT_TYPES:
         raise ValueError(
-            f'filter_output should be one of {", ".join(FILTER_FACTORY.keys())}, received {filter_arg}'
+            f'filter_output should be one of {", ".join(FILTER_OUTPUT_TYPES)}, received {filter_arg}'
         )
 
 

Original file line number	Diff line number	Diff line change
`@@ -54,9 +54,9 @@`
`54`	`54`
`55`	`55`
`56`	`56`	`def _validate_filter_arg(filter_arg):`
`57`		`- if filter_arg.lower() not in FILTER_FACTORY.keys():`
	`57`	`+ if filter_arg.lower() not in FILTER_OUTPUT_TYPES:`
`58`	`58`	`raise ValueError(`
`59`		`- f'filter_output should be one of {", ".join(FILTER_FACTORY.keys())}, received {filter_arg}'`
	`59`	`+ f'filter_output should be one of {", ".join(FILTER_OUTPUT_TYPES)}, received {filter_arg}'`
`60`	`60`	`)`
`61`	`61`
`62`	`62`