save_to_df testing

This commit is contained in:
Mayank Aamseek
2024-04-09 16:59:31 +05:30
parent 6b66839dd4
commit 428afcb6ad
+7 -7
View File
@@ -64,14 +64,14 @@ def save_to_sf(dag_name, **kwargs):
# attempt_logs_file = "attempt_logs_sample.csv"
# conf = "{\"" + "training_results_file_name" + "\":\"" + {training_results_file} + "\", \"" + "attempt_logs_file_name" + "\":\"" + {attempt_logs_file} + "\"}".format(training_results_file=training_results_file, attempt_logs_file=attempt_logs_file)
conf = f"""{{
"{list(kwargs.keys())[0]}": "{list(kwargs.values())[0]}",
"{list(kwargs.keys())[1]}": "{list(kwargs.values())[1]}"
}}"""
# conf = f"""{{
# "{list(kwargs.keys())[0]}": "{list(kwargs.values())[0]}",
# "{list(kwargs.keys())[1]}": "{list(kwargs.values())[1]}"
# }}"""
# conf = f"""
# {{ {json.dumps(kwargs)}
# }}"""
conf = f"""
{{ {json.dumps(kwargs)}
}}"""
payload = mwaa_cli_command + " " + dag_name + " --conf '{}'".format(conf)
headers = {