From 2e7bf4a0d5c8fa0a145f4995a48eeae46637c884 Mon Sep 17 00:00:00 2001 From: Mary Brown Date: Tue, 16 Jun 2026 12:46:10 +0000 Subject: [PATCH] update with price band --- src/template_code_location/ops/ops.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/template_code_location/ops/ops.py b/src/template_code_location/ops/ops.py index e929ce1..83826c7 100644 --- a/src/template_code_location/ops/ops.py +++ b/src/template_code_location/ops/ops.py @@ -32,5 +32,9 @@ 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