fix: SupplementaryTextClassifier failure

alphanome-ai · Dec 21, 2023 · cc1d1d7 · cc1d1d7
1 parent d7995e8
commit cc1d1d7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sec_parser/processing_engine/core.py b/sec_parser/processing_engine/core.py
@@ -180,13 +180,13 @@ def get_default_steps(
             IntroductorySectionElementClassifier(),
             TextClassifier(types_to_process={NotYetClassifiedElement}),
             HighlightedTextClassifier(types_to_process={TextElement}),
-            PageHeaderClassifier(
+            SupplementaryTextClassifier(
                 types_to_process={TextElement, HighlightedTextElement},
             ),
-            PageNumberClassifier(
+            PageHeaderClassifier(
                 types_to_process={TextElement, HighlightedTextElement},
             ),
-            SupplementaryTextClassifier(
+            PageNumberClassifier(
                 types_to_process={TextElement, HighlightedTextElement},
             ),
             TitleClassifier(types_to_process={HighlightedTextElement}),