cmcmaster commited on
Commit
9eba3ac
·
verified ·
1 Parent(s): dc82736

Update pbs_data.py

Browse files
Files changed (1) hide show
  1. pbs_data.py +2 -2
pbs_data.py CHANGED
@@ -411,14 +411,14 @@ class PBSPublicDataAPIClient:
411
 
412
  return processed
413
 
414
- def save_data_to_hf(self, data, dataset_name="cmcmaster/rheumatology-biologics-dataset"):
415
  processed_data = self.preprocess_data(data)
416
 
417
  # Create a Dataset from the combinations
418
  dataset = Dataset.from_list(processed_data['combinations'])
419
 
420
  # Push the dataset to the Hugging Face Hub
421
- dataset.push_to_hub(dataset_name)
422
 
423
  print(f"Data saved to Hugging Face Hub: {dataset_name}")
424
 
 
411
 
412
  return processed
413
 
414
+ def save_data_to_hf(self, data, hf_token, dataset_name="cmcmaster/rheumatology-biologics-dataset"):
415
  processed_data = self.preprocess_data(data)
416
 
417
  # Create a Dataset from the combinations
418
  dataset = Dataset.from_list(processed_data['combinations'])
419
 
420
  # Push the dataset to the Hugging Face Hub
421
+ dataset.push_to_hub(dataset_name, token=hf_token)
422
 
423
  print(f"Data saved to Hugging Face Hub: {dataset_name}")
424