attila-balint-kul commited on
Commit
54af163
·
verified ·
1 Parent(s): 4cc09e5

Update components.py

Browse files
Files changed (1) hide show
  1. components.py +2 -2
components.py CHANGED
@@ -73,7 +73,7 @@ def model_selector(models: list[str], data: pd.DataFrame) -> set[str]:
73
  # Group models by their prefix
74
  model_groups: dict[str, list[str]] = {}
75
  for model in models:
76
- group, model_name = model.split(".", maxsplit=1)
77
  if group not in model_groups:
78
  model_groups[group] = []
79
  model_groups[group].append(model_name)
@@ -353,7 +353,7 @@ def models_view(data: pd.DataFrame):
353
  with right:
354
  st.markdown("#### Frameworks")
355
  _df = models.copy()
356
- _df["Framework"] = _df.index.str.split(".").str[0]
357
  fig = px.pie(
358
  _df.groupby("Framework").size().reset_index(),
359
  values=0,
 
73
  # Group models by their prefix
74
  model_groups: dict[str, list[str]] = {}
75
  for model in models:
76
+ group, model_name = model.split("-", maxsplit=1)
77
  if group not in model_groups:
78
  model_groups[group] = []
79
  model_groups[group].append(model_name)
 
353
  with right:
354
  st.markdown("#### Frameworks")
355
  _df = models.copy()
356
+ _df["Framework"] = _df.index.str.split("-").str[0]
357
  fig = px.pie(
358
  _df.groupby("Framework").size().reset_index(),
359
  values=0,