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 diff --git a/src/template_code_location/repository.py b/src/template_code_location/repository.py index 5e985d0..2aa18e9 100644 --- a/src/template_code_location/repository.py +++ b/src/template_code_location/repository.py @@ -9,10 +9,11 @@ from template_code_location.jobs.jobs import (data_processing_job, from util_services.custom_json_logger import simpl_json_logger from util_services.resources import s3_resource + defs = Definitions( jobs=[ - example_job, data_processing_job, + example_dataframe_demo_job, k_anonymity_job_s3, l_diversity_job_s3, t_closeness_job_s3,