refactor: clean up imports and improve gender normalization method

2025-09-20 22:55:24 +02:00
parent 0816207a2c
commit dd2a9f2711
5 changed files with 226 additions and 82 deletions
@@ -150,7 +150,8 @@ class FeatureExtractionStep(PipelineStep):
            except Exception as e:
                logging.warning(f"NER tagging failed for row {idx}: {e}")

-    def _normalize_gender(self, series: pd.Series) -> pd.Series:
+    @classmethod
+    def _normalize_gender(cls, series: pd.Series) -> pd.Series:
        gender_mapping = {
            "m": "m",
            "male": "m",