PVNet 5.0.10__tar.gz → 5.0.12__tar.gz

{pvnet-5.0.10 → pvnet-5.0.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PVNet
-Version: 5.0.10
+Version: 5.0.12
 Summary: PVNet
 Author-email: Peter Dudfield <info@openclimatefix.org>
 Requires-Python: >=3.11

{pvnet-5.0.10 → pvnet-5.0.12}/PVNet.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PVNet
-Version: 5.0.10
+Version: 5.0.12
 Summary: PVNet
 Author-email: Peter Dudfield <info@openclimatefix.org>
 Requires-Python: >=3.11

{pvnet-5.0.10 → pvnet-5.0.12}/pvnet/models/base_model.py

@@ -129,6 +129,7 @@ def download_from_hf(
     force_download: bool,
     max_retries: int = 5,
     wait_time: int = 10,
+    token: bool | str | None = None,
 ) -> str | list[str]:
     """Tries to download one or more files from HuggingFace up to max_retries times.
 
@@ -140,6 +141,9 @@ def download_from_hf(
         force_download: Whether to force a new download
         max_retries: Maximum number of retry attempts
         wait_time: Wait time (in seconds) before retrying
+        token: 
+            HF authentication token. If True, the token is read from the HuggingFace config folder.
+            If a string, it is used as the authentication token. 
 
     Returns:
         The local file path of the downloaded file(s)
@@ -152,6 +156,7 @@ def download_from_hf(
                 revision=revision,
                 cache_dir=cache_dir,
                 force_download=force_download,
+                token=token,
             )
 
             if isinstance(filename, list):
@@ -220,6 +225,7 @@ class HuggingfaceMixin:
         revision: str,
         cache_dir: str | None = None,
         force_download: bool = False,
+        token: bool | str | None = None,
     ) -> str:
         """Load data config file."""
         if os.path.isdir(model_id):
@@ -235,6 +241,7 @@ class HuggingfaceMixin:
                 force_download=force_download,
                 max_retries=5,
                 wait_time=10,
+                token=token
             )
 
         return data_config_file