Malikeh Ehghaghi
commited on
Commit
•
b424024
1
Parent(s):
872190a
Update dataset_list.py
Browse files- dataset_list.py +1 -6
dataset_list.py
CHANGED
|
@@ -66,21 +66,16 @@ class DatasetList:
|
|
| 66 |
df = df[df.dataset_name.str.contains(search_query)]
|
| 67 |
else:
|
| 68 |
df = df[df.dataset_name_lowercase.str.contains(search_query.lower())]
|
| 69 |
-
has_dataset = 'Dataset' in filter_names
|
| 70 |
has_datalink = 'Data Link' in filter_names
|
| 71 |
has_paper = 'Paper' in filter_names
|
| 72 |
-
df = self.filter_table(df,
|
| 73 |
-
#df = self.filter_table(df, has_paper, has_github, has_model, data_types, model_types)
|
| 74 |
return len(df), self.to_html(df, self.table_header)
|
| 75 |
|
| 76 |
@staticmethod
|
| 77 |
def filter_table(df: pd.DataFrame,
|
| 78 |
-
has_dataset: bool,
|
| 79 |
has_datalink: bool,
|
| 80 |
has_paper: bool
|
| 81 |
) -> pd.DataFrame:
|
| 82 |
-
# if has_dataset:
|
| 83 |
-
# df = df[~df.dataset_name.isna()]
|
| 84 |
if has_datalink:
|
| 85 |
df = df[~df.huggingface_link.isna() | ~df.original_link.isna()]
|
| 86 |
if has_paper:
|
|
|
|
| 66 |
df = df[df.dataset_name.str.contains(search_query)]
|
| 67 |
else:
|
| 68 |
df = df[df.dataset_name_lowercase.str.contains(search_query.lower())]
|
|
|
|
| 69 |
has_datalink = 'Data Link' in filter_names
|
| 70 |
has_paper = 'Paper' in filter_names
|
| 71 |
+
df = self.filter_table(df, has_datalink, has_paper)
|
|
|
|
| 72 |
return len(df), self.to_html(df, self.table_header)
|
| 73 |
|
| 74 |
@staticmethod
|
| 75 |
def filter_table(df: pd.DataFrame,
|
|
|
|
| 76 |
has_datalink: bool,
|
| 77 |
has_paper: bool
|
| 78 |
) -> pd.DataFrame:
|
|
|
|
|
|
|
| 79 |
if has_datalink:
|
| 80 |
df = df[~df.huggingface_link.isna() | ~df.original_link.isna()]
|
| 81 |
if has_paper:
|