diff --git a/src/delta_barth/analysis/parse.py b/src/delta_barth/analysis/parse.py index bec6308..79b0a83 100644 --- a/src/delta_barth/analysis/parse.py +++ b/src/delta_barth/analysis/parse.py @@ -3,7 +3,7 @@ from __future__ import annotations from collections.abc import Mapping, Set from typing import TYPE_CHECKING -from delta_barth.errors import FeaturesMissingError +from delta_barth.errors import FeaturesMissingError, wrap_result if TYPE_CHECKING: import pandas as pd @@ -51,3 +51,12 @@ def preprocess_features( _check_needed_features(data, target_features) return data + + +@wrap_result() +def preprocess_features_wrapped( + data: pd.DataFrame, + feature_map: Mapping[str, str], + target_features: Set[str], +) -> pd.DataFrame: # pragma: no cover + return preprocess_features(data, feature_map, target_features)