Revert a few more changes

pymc-devs · Dec 12, 2024 · 58bc697 · 58bc697
1 parent a8a8a6b
commit 58bc697
Showing 1 changed file with 2 additions and 2 deletions.
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}'
         )