alfraser commited on
Commit
2122072
1 Parent(s): c0848c6

Switched endpoint control to use the writeable token as it was inconsistent with the normal token.

Browse files
Files changed (1) hide show
  1. src/common.py +2 -2
src/common.py CHANGED
@@ -80,14 +80,14 @@ def hf_endpoint_status(username: str, endpoint_name: str, api_token: str = None)
80
  def pause_hf_endpoint(username: str, endpoint_name: str, api_token: str = None) -> None:
81
  url = f'https://api.endpoints.huggingface.cloud/v2/endpoint/{username}/{endpoint_name}/pause'
82
  if api_token is None:
83
- api_token = hf_api_token()
84
  requests.post(url, headers={"Authorization": f"Bearer {api_token}"})
85
 
86
 
87
  def resume_hf_endpoint(username: str, endpoint_name: str, api_token: str = None) -> None:
88
  url = f'https://api.endpoints.huggingface.cloud/v2/endpoint/{username}/{endpoint_name}/resume'
89
  if api_token is None:
90
- api_token = hf_api_token()
91
  requests.post(url, headers={"Authorization": f"Bearer {api_token}"})
92
 
93
 
 
80
  def pause_hf_endpoint(username: str, endpoint_name: str, api_token: str = None) -> None:
81
  url = f'https://api.endpoints.huggingface.cloud/v2/endpoint/{username}/{endpoint_name}/pause'
82
  if api_token is None:
83
+ api_token = hf_api_token(write=True)
84
  requests.post(url, headers={"Authorization": f"Bearer {api_token}"})
85
 
86
 
87
  def resume_hf_endpoint(username: str, endpoint_name: str, api_token: str = None) -> None:
88
  url = f'https://api.endpoints.huggingface.cloud/v2/endpoint/{username}/{endpoint_name}/resume'
89
  if api_token is None:
90
+ api_token = hf_api_token(write=True)
91
  requests.post(url, headers={"Authorization": f"Bearer {api_token}"})
92
 
93