From f611731a188bf56545a7eb7fce3801cb9935e8d8 Mon Sep 17 00:00:00 2001
From: Dan Saattrup Nielsen <dan.nielsen@alexandra.dk>
Date: Mon, 31 Jul 2023 15:17:36 +0200
Subject: [PATCH] chore: Try using `use_auth_token` instead of `token`

---
 src/coral_models/data.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/coral_models/data.py b/src/coral_models/data.py
index ca3144e8..b54864e7 100644
--- a/src/coral_models/data.py
+++ b/src/coral_models/data.py
@@ -29,7 +29,7 @@ def load_data(cfg: DictConfig) -> DatasetDict | IterableDatasetDict:
     dataset = load_dataset(
         path=cfg.dataset.id,
         name=cfg.dataset.subset,
-        token=os.getenv("HUGGINGFACE_HUB_TOKEN", True),
+        use_auth_token=os.getenv("HUGGINGFACE_HUB_TOKEN", True),
         streaming=True,
     )