Don't re-login every time

PolicyEngine · Nov 29, 2024 · a83fbc8 · a83fbc8
1 parent 4bdebb0
commit a83fbc8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/policyengine_core/data/dataset.py b/policyengine_core/data/dataset.py
@@ -461,14 +461,14 @@ def upload_to_huggingface(self, owner_name: str, model_name: str):
         token = os.environ.get(
             "HUGGING_FACE_TOKEN",
         )
-        login(token=token)
         api = HfApi()
 
         api.upload_file(
             path_or_fileobj=self.file_path,
             path_in_repo=self.file_path.name,
             repo_id=f"{owner_name}/{model_name}",
             repo_type="model",
+            token=token,
         )
 
     def download_from_huggingface(
@@ -483,11 +483,11 @@ def download_from_huggingface(
         token = os.environ.get(
             "HUGGING_FACE_TOKEN",
         )
-        login(token=token)
 
         hf_hub_download(
             repo_id=f"{owner_name}/{model_name}",
             repo_type="model",
             path=self.file_path,
             revision=version,
+            token=token,
         )