def __init__()

in src/preprocess.py [0:0]

• 24 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self, input_data) -> None:
        self._input_data = input_data
        self._logger.debug("Defining transformers.")
        numeric_features = list(feature_columns_names)
        numeric_features.remove("sex")
        numeric_transformer = Pipeline(
            steps=[("imputer", SimpleImputer(strategy="median")), ("scaler", StandardScaler())]
        )

        categorical_features = ["sex"]
        categorical_transformer = Pipeline(
            steps=[
                ("imputer", SimpleImputer(strategy="constant", fill_value="missing")),
                ("onehot", OneHotEncoder(handle_unknown="ignore")),
            ]
        )

        self._preprocess = ColumnTransformer(
            transformers=[
                ("num", numeric_transformer, numeric_features),
                ("cat", categorical_transformer, categorical_features),
            ]
        )

        self._logger.debug("Fitting transforms.")
        self._input_data_y = self._input_data.pop("rings")
        self._preprocess.fit(self._input_data)