Adding "Unknown" if num. params is None

#9
by fzoll - opened
Files changed (1) hide show
  1. app/backend/data_engine.py +4 -0
app/backend/data_engine.py CHANGED
@@ -104,6 +104,10 @@ class DataEngine:
104
  models_list = self.models
105
 
106
  df_model = pd.DataFrame(models_list)
 
 
 
 
107
 
108
  # Create mapping for model names/aliases
109
  if 'alias' in df_model.columns:
 
104
  models_list = self.models
105
 
106
  df_model = pd.DataFrame(models_list)
107
+
108
+ # Replace None values in num_params with "Unknown"
109
+ if 'num_params' in df_model.columns:
110
+ df_model['num_params'] = df_model['num_params'].fillna("Unknown")
111
 
112
  # Create mapping for model names/aliases
113
  if 'alias' in df_model.columns: