From fbc4df133c61eaaf5d207f604c0e40f799cad8b7 Mon Sep 17 00:00:00 2001
From: Marco Castelluccio <mcastelluccio@mozilla.com>
Date: Tue, 17 Feb 2026 18:25:31 +0100
Subject: [PATCH] Fit the extraction pipeline before applying it

---
 bugbug/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/bugbug/model.py b/bugbug/model.py
index 9ef60c51d3..fbae7595b7 100644
--- a/bugbug/model.py
+++ b/bugbug/model.py
@@ -370,7 +370,7 @@ def train(self, importance_cutoff=0.15, limit=None):
         X_gen, y = split_tuple_generator(lambda: self.items_gen(classes))
 
         # Extract features from the items.
-        X = self.extraction_pipeline.transform(X_gen)
+        X = self.extraction_pipeline.fit_transform(X_gen)
 
         # Calculate labels.
         y = np.array(y)