From be3c2710418853b6c67894ec5c31a9f467bb7610 Mon Sep 17 00:00:00 2001 From: Mary Brown Date: Mon, 15 Jun 2026 13:08:59 +0000 Subject: [PATCH] Prep --- src/template_code_location/ops/ops.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/template_code_location/ops/ops.py b/src/template_code_location/ops/ops.py index 994c18b..3a73b18 100644 --- a/src/template_code_location/ops/ops.py +++ b/src/template_code_location/ops/ops.py @@ -31,8 +31,7 @@ def generate_example_dataset() -> pd.DataFrame: def transform_example_dataset(df: pd.DataFrame) -> pd.DataFrame: """Applies a simple transformation to the example dataset.""" - df["price_band"] = df["total_price"].apply( - lambda value: "high" if value >= 25 else "standard" - ) + df["total_price"] = df["quantity"] * df["unit_price"] + return df