mridzuan commited on
Commit
47c68ae
·
1 Parent(s): 0b1b00d

fix syntax error

Browse files
Files changed (1) hide show
  1. src/model_utils.py +2 -2
src/model_utils.py CHANGED
@@ -70,7 +70,7 @@ def save_model(model, user_model_name, metrics_result_single=None):
70
  login(token=os.environ["HF_TOKEN"])
71
 
72
  timestamp = datetime.now().strftime("%y%m%d_%H%M%S")
73
- filename = f"{timestamp}{st.session_state.get("target_col", "UNKNOWN")[0]}_{user_model_name}_single"
74
 
75
  # Prepare model dict (same as before)
76
  model_data = {
@@ -142,7 +142,7 @@ def save_model_ensemble(models, user_model_name, best_iterations=None, fold_scor
142
  login(token=os.environ["HF_TOKEN"])
143
 
144
  timestamp = datetime.now().strftime("%y%m%d_%H%M%S")
145
- filename = f"{timestamp}{st.session_state.get("target_col", "UNKNOWN")[0]}_{user_model_name}_ensemble"
146
 
147
  ensemble_data = {
148
  "timestamp": timestamp,
 
70
  login(token=os.environ["HF_TOKEN"])
71
 
72
  timestamp = datetime.now().strftime("%y%m%d_%H%M%S")
73
+ filename = f"{timestamp}{st.session_state.get('target_col', 'UNKNOWN')[0]}_{user_model_name}_single"
74
 
75
  # Prepare model dict (same as before)
76
  model_data = {
 
142
  login(token=os.environ["HF_TOKEN"])
143
 
144
  timestamp = datetime.now().strftime("%y%m%d_%H%M%S")
145
+ filename = f"{timestamp}{st.session_state.get('target_col', 'UNKNOWN')[0]}_{user_model_name}_ensemble"
146
 
147
  ensemble_data = {
148
  "timestamp": timestamp,