hysts HF staff commited on
Commit
0d10e90
1 Parent(s): fee55b3
Files changed (1) hide show
  1. paper_list.py +9 -7
paper_list.py CHANGED
@@ -71,9 +71,10 @@ class PaperList:
71
  has_hf_dataset = 'HF Dataset' in filter_names
72
  df = self.filter_table(df, has_supp, has_arxiv, has_github,
73
  has_hf_space, has_hf_model, has_hf_dataset)
74
- return len(df), self.to_html(df)
75
 
76
- def filter_table(self, df: pd.DataFrame, has_supp: bool, has_arxiv: bool,
 
77
  has_github: bool, has_hf_space: bool, has_hf_model: bool,
78
  has_hf_dataset: bool) -> pd.DataFrame:
79
  if has_supp:
@@ -90,11 +91,12 @@ class PaperList:
90
  df = df[~df.hf_dataset.isna()]
91
  return df
92
 
93
- def to_html(self, df: pd.DataFrame) -> str:
94
- table_rows = df.html_table_content
95
- table_data = ''.join(table_rows)
96
- html = f'''<table>
97
- {self.table_header}
 
98
  {table_data}
99
  </table>'''
100
  return html
 
71
  has_hf_dataset = 'HF Dataset' in filter_names
72
  df = self.filter_table(df, has_supp, has_arxiv, has_github,
73
  has_hf_space, has_hf_model, has_hf_dataset)
74
+ return len(df), self.to_html(df, self.table_header)
75
 
76
+ @staticmethod
77
+ def filter_table(df: pd.DataFrame, has_supp: bool, has_arxiv: bool,
78
  has_github: bool, has_hf_space: bool, has_hf_model: bool,
79
  has_hf_dataset: bool) -> pd.DataFrame:
80
  if has_supp:
 
91
  df = df[~df.hf_dataset.isna()]
92
  return df
93
 
94
+ @staticmethod
95
+ def to_html(df: pd.DataFrame, table_header: str) -> str:
96
+ table_data = ''.join(df.html_table_content)
97
+ html = f'''
98
+ <table>
99
+ {table_header}
100
  {table_data}
101
  </table>'''
102
  return html