From 035217023dddaf533023ebc7e972b54ae0a445b6 Mon Sep 17 00:00:00 2001 From: Mayank Aamseek Date: Tue, 9 Apr 2024 19:37:36 +0530 Subject: [PATCH] save_to_df testing --- streamlit/sf_conn.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/streamlit/sf_conn.py b/streamlit/sf_conn.py index 12b3246..69683fd 100644 --- a/streamlit/sf_conn.py +++ b/streamlit/sf_conn.py @@ -39,7 +39,7 @@ def get_secret(): # TODO: This function needs to be changed to accept Kwargs # The function name should be more generic and cofnigurable -def save_to_sf(dag_name, conf_1, conf_2, training_results_file, attempt_logs_file): +def save_to_sf(dag_name, dict_args): mwaa_env_name = 'doczy-dev-infra-mwaa' dag_name = dag_name @@ -62,6 +62,10 @@ def save_to_sf(dag_name, conf_1, conf_2, training_results_file, attempt_logs_fil # 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_1 = list(dict_args.keys())[0] + conf_2 = list(dict_args.keys())[1] + training_results_file = list(dict_args.values())[0] + training_results_file = list(dict_args.values())[1] conf = f"""{{ "{conf_1}": "{training_results_file}", "{conf_2}": "{attempt_logs_file}"