clean - remove 'pkl'

src/model_utils.py

@@ -60,7 +60,7 @@ def save_model(model, user_model_name, metrics_result_single=None):
         login(token=os.environ["HF_TOKEN"])
 
     timestamp = datetime.now().strftime("%y%m%d_%H%M%S")
-    filename = f"{timestamp}_{user_model_name}_single.pkl"
+    filename = f"{timestamp}_{user_model_name}_single"
 
     # Prepare model dict (same as before)
     model_data = {
@@ -112,7 +112,7 @@ def save_model(model, user_model_name, metrics_result_single=None):
     scheduler.api.upload_file(
         repo_id=os.environ["HF_REPO_ID"],
         repo_type="dataset",
-        path_in_repo=f"models/{filename.replace('.pkl', '')}.parquet",
+        path_in_repo=f"models/{filename}.parquet",
         path_or_fileobj=buf
     )
 
@@ -132,7 +132,7 @@ def save_model_ensemble(models, user_model_name, best_iterations=None, fold_scor
         login(token=os.environ["HF_TOKEN"])
 
     timestamp = datetime.now().strftime("%y%m%d_%H%M%S")
-    filename = f"{timestamp}_{user_model_name}_ensemble.pkl"
+    filename = f"{timestamp}_{user_model_name}_ensemble"
 
     ensemble_data = {
         "timestamp": timestamp,
@@ -180,7 +180,7 @@ def save_model_ensemble(models, user_model_name, best_iterations=None, fold_scor
     scheduler.api.upload_file(
         repo_id=os.environ["HF_REPO_ID"],
         repo_type="dataset",
-        path_in_repo=f"models/{filename.replace('.pkl', '')}.parquet",
+        path_in_repo=f"models/{filename}.parquet",
         path_or_fileobj=buf
     )
 
@@ -212,7 +212,7 @@ def load_model(model_name):
         )
         table = pq.read_table(downloaded)
         row = table.to_pylist()[0]
-        if row["filename"] == model_name.replace("parquet", "pkl"):
+        if row["filename"] == model_name.replace("parquet", ""):
             target_file = downloaded
             break
 

src/sidebar.py

@@ -25,7 +25,7 @@ def sidebar():
                 )
                 table = pq.read_table(downloaded)
                 row = table.to_pylist()[0]
-                models.append(row["filename"].replace(".parquet","").replace(".pkl",""))
+                models.append(row["filename"].replace(".parquet",""))
             except Exception as e:
                 st.warning(f"Skipping model file due to error: {f} ({e})")