diff --git a/src/template_code_location/ops/ops.py b/src/template_code_location/ops/ops.py index eb576e8..e929ce1 100644 --- a/src/template_code_location/ops/ops.py +++ b/src/template_code_location/ops/ops.py @@ -32,9 +32,5 @@ def transform_example_dataset(df: pd.DataFrame) -> pd.DataFrame: """Applies a simple transformation to the example dataset.""" df["total_price"] = df["quantity"] * df["unit_price"] - - df["price_band"] = df["total_price"].apply( - lambda value: "high" if value >= 25 else "standard" - ) - + return df