davanstrien HF staff commited on
Commit
e96101e
·
1 Parent(s): e6064e2

chore: Update data_loader.py with Hugging Face Hub integration

Browse files
Files changed (1) hide show
  1. data_loader.py +2 -1
data_loader.py CHANGED
@@ -20,6 +20,7 @@ assert (
20
  logger = logging.get_logger(__name__)
21
 
22
  api = HfApi(token=HF_TOKEN)
 
23
 
24
 
25
  def has_card_data(dataset):
@@ -95,7 +96,7 @@ def refresh_data() -> List[Dict[str, Any]]:
95
  return df.to_dict(orient="records")
96
 
97
  # List all datasets
98
- datasets = list(api.list_datasets(limit=None, full=True))
99
 
100
  # Filter datasets with dataset info
101
  datasets = [
 
20
  logger = logging.get_logger(__name__)
21
 
22
  api = HfApi(token=HF_TOKEN)
23
+ MAX_DATASETS = 1000
24
 
25
 
26
  def has_card_data(dataset):
 
96
  return df.to_dict(orient="records")
97
 
98
  # List all datasets
99
+ datasets = list(api.list_datasets(limit=MAX_DATASETS, full=True))
100
 
101
  # Filter datasets with dataset info
102
  datasets = [