[model:backout] Use the correct pipeline type (#3894)

mozilla · Dec 5, 2023 · c9fdb64 · c9fdb64
1 parent 6b0a6d3
commit c9fdb64
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bugbug/models/backout.py b/bugbug/models/backout.py
@@ -74,7 +74,7 @@ def __init__(self, lemmatization=False, bug_data=False):
             feature_cleanup.synonyms(),
         ]
 
-        self.extraction_pipeline = ImblearnPipeline(
+        self.extraction_pipeline = Pipeline(
             [
                 (
                     "commit_extractor",
@@ -85,7 +85,7 @@ def __init__(self, lemmatization=False, bug_data=False):
             ]
         )
 
-        self.clf = Pipeline(
+        self.clf = ImblearnPipeline(
             [
                 (
                     "union",