From 8eb3586921b822e6e07e2c543da998b6d9a12d90 Mon Sep 17 00:00:00 2001 From: Michael Wagner Date: Fri, 6 Dec 2024 20:38:59 +0000 Subject: [PATCH] Add updated run_type logic --- manual_flagging/flagging.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/manual_flagging/flagging.py b/manual_flagging/flagging.py index d1bfbeb..0409f43 100755 --- a/manual_flagging/flagging.py +++ b/manual_flagging/flagging.py @@ -427,13 +427,18 @@ def create_bins_and_labels(input_list): # Get sale.metadata table commit_sha = sp.getoutput("git rev-parse HEAD") -# Write to sale.group_mean table +run_type = ( + "initial_flagging" + if not inputs["manual_update"] + else "manual_update_only_new_sales" + if inputs["manual_update_only_new_sales"] + else "manual_update" +) + df_metadata = flg.get_metadata_df( run_id=run_id, timestamp=timestamp, - run_type="initial_flagging" - if inputs["manual_update"] == False - else "manual_update", + run_type=run_type, commit_sha=commit_sha, run_note=inputs["run_note"], )