MNT Rename the env variable HF_ENDPOINT. (#819)

* Rename the env variable `HF_ENDPOINT`. * Reuse `ENDPOINT` constant to reduce name clashes. * Quality.
huggingface · Apr 5, 2022 · bee7f64 · bee7f64
1 parent 912078f
commit bee7f64
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/src/huggingface_hub/constants.py b/src/huggingface_hub/constants.py
@@ -24,7 +24,7 @@
     os.environ.get("HUGGINGFACE_CO_STAGING", "NO").upper() in ENV_VARS_TRUE_VALUES
 )
 
-ENDPOINT = os.getenv("HF_HUB_URL") or (
+ENDPOINT = os.getenv("HF_ENDPOINT") or (
     "https://moon-staging.huggingface.co" if _staging_mode else "https://huggingface.co"
 )
 

diff --git a/src/huggingface_hub/hf_api.py b/src/huggingface_hub/hf_api.py
@@ -466,9 +466,7 @@ def erase_from_credential_store(username=None):
 
 class HfApi:
     def __init__(self, endpoint=None):
-        self.endpoint = (
-            endpoint if endpoint is not None else os.getenv("HF_ENDPOINT", ENDPOINT)
-        )
+        self.endpoint = endpoint if endpoint is not None else ENDPOINT
 
     def login(self, username: str, password: str) -> str:
         """