katielink commited on
Commit
d63a255
·
1 Parent(s): 0a4ba8c

Update model_list.py

Browse files
Files changed (1) hide show
  1. model_list.py +5 -2
model_list.py CHANGED
@@ -73,7 +73,10 @@ class ModelList:
73
 
74
  @staticmethod
75
  def filter_table(df: pd.DataFrame, has_paper: bool, has_github: bool,
76
- has_model: bool, data_types: list[str], model_types: list[str]) -> pd.DataFrame:
 
 
 
77
  if has_paper:
78
  df = df[~df.paper.isna()]
79
  if has_github:
@@ -81,7 +84,7 @@ class ModelList:
81
  if has_model:
82
  df = df[~df.hub.isna() | ~df.other.isna()]
83
  df = df[df.data_type.isin(set(data_types))]
84
- df = df[df.base_model.isin(set(model_types))]
85
  return df
86
 
87
  @staticmethod
 
73
 
74
  @staticmethod
75
  def filter_table(df: pd.DataFrame, has_paper: bool, has_github: bool,
76
+ has_model: bool,
77
+ data_types: list[str],
78
+ #model_types: list[str]
79
+ ) -> pd.DataFrame:
80
  if has_paper:
81
  df = df[~df.paper.isna()]
82
  if has_github:
 
84
  if has_model:
85
  df = df[~df.hub.isna() | ~df.other.isna()]
86
  df = df[df.data_type.isin(set(data_types))]
87
+ #df = df[df.base_model.isin(set(model_types))]
88
  return df
89
 
90
  @staticmethod