fix: Fixed transformations

src/data.py

@@ -159,18 +159,25 @@ def preprocess_data(
         (f"hhmm_{feature}", dtf.hhmm_transformer(feature)) for feature in cfg.hhmm
     ]
 
+    # Assembling the pipeline
+    pipeline = Pipeline(
+        [
+            feature_extractor,
+            *cyclic_transformers,
+            *hhmm_transformers,
+        ]
+    )
+
+    X = pipeline.transform(X)
+
     # Transformers for hashing
     hash_transformers = [
         (f"hash_{feature}", dtf.hash_transformer(feature))
-        for feature in df.columns[df.dtypes == "object"]
+        for feature in X.columns[X.dtypes == "object"]
     ]
 
-    # Assembling the pipeline
     pipeline = Pipeline(
         [
-            feature_extractor,
-            *cyclic_transformers,
-            *hhmm_transformers,
             *hash_transformers,
         ]
     )

src/data_transformations.py

@@ -12,7 +12,7 @@ def __init__(self, features: list[str]):
     def transform(self, X):
         columns_to_drop = set(X) - set(self.features)
 
-        X_transformed = X[list[columns_to_drop]]
+        X_transformed = X.drop(columns_to_drop, axis=1)
 
         return X_transformed