tathagataraha commited on
Commit
2e9477a
·
1 Parent(s): b50c184

[MODIFY] Disabled model size filter for private

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -159,8 +159,11 @@ def update_table(
159
  show_deleted: bool = False,
160
  ):
161
  filtered_df = filter_models(hidden_df, type_query, domain_specific_query, size_query, precision_query, show_deleted)
 
162
  filtered_df = filter_queries(query, filtered_df)
 
163
  df = select_columns(filtered_df, columns, list(hidden_df.columns))
 
164
  return df
165
 
166
 
@@ -233,7 +236,7 @@ def filter_models(
233
  if AutoEvalColumn.precision.name in df.columns:
234
  filtered_df = filtered_df.loc[df[AutoEvalColumn.precision.name].isin(precision_query + ["None"])]
235
 
236
- if size_query is not None:
237
  numeric_interval = pd.IntervalIndex(sorted([NUMERIC_INTERVALS[s] for s in size_query]))
238
  params_column = pd.to_numeric(df[AutoEvalColumn.params.name], errors="coerce")
239
  mask = params_column.apply(lambda x: any(numeric_interval.contains(x)))
 
159
  show_deleted: bool = False,
160
  ):
161
  filtered_df = filter_models(hidden_df, type_query, domain_specific_query, size_query, precision_query, show_deleted)
162
+ # breakpoint()
163
  filtered_df = filter_queries(query, filtered_df)
164
+ # breakpoint()
165
  df = select_columns(filtered_df, columns, list(hidden_df.columns))
166
+ # breakpoint()
167
  return df
168
 
169
 
 
236
  if AutoEvalColumn.precision.name in df.columns:
237
  filtered_df = filtered_df.loc[df[AutoEvalColumn.precision.name].isin(precision_query + ["None"])]
238
 
239
+ if size_query is not None and not PRIVATE_REPO:
240
  numeric_interval = pd.IntervalIndex(sorted([NUMERIC_INTERVALS[s] for s in size_query]))
241
  params_column = pd.to_numeric(df[AutoEvalColumn.params.name], errors="coerce")
242
  mask = params_column.apply(lambda x: any(numeric_interval.contains(x)))