File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -59,6 +59,8 @@ def merge_data(tables: dict) -> pd.DataFrame:
5959 df_payments , on = "order_id" , how = "left"
6060 )
6161
62+ df_merged = df_merged .rename (columns = {"payment_value" : "order_revenue" })
63+
6264 if len (df_merged ) != len (df_orders ):
6365 raise RuntimeError ("Cardinality violation detected: expected 1 row per order" )
6466
@@ -136,6 +138,7 @@ def freeze_schema(df: pd.DataFrame) -> pd.DataFrame:
136138
137139 ENFORCED_SCHEMA = [
138140 "order_id" ,
141+ "order_revenue" ,
139142 "seller_id" ,
140143 "product_id" ,
141144 "order_status" ,
@@ -153,6 +156,7 @@ def freeze_schema(df: pd.DataFrame) -> pd.DataFrame:
153156
154157 ENFORCED_DTYPES = {
155158 "order_id" : "string" ,
159+ "order_revenue" : "float64" ,
156160 "seller_id" : "string" ,
157161 "product_id" : "string" ,
158162 "order_status" : "string" ,
You can’t perform that action at this time.
0 commit comments