Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-24 03:59:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix test

Browse Source
This commit is contained in:
Yingda Chen
2025-03-04 16:44:42 +08:00
parent 18fb4286d9
commit f8dc0045e2
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
modelscope/hub/api.py
View File

@@ -17,6 +17,7 @@ from http.cookiejar import CookieJar
from os.path import expanduser from os.path import expanduser
from pathlib import Path from pathlib import Path
from typing import Any, BinaryIO, Dict, Iterable, List, Optional, Tuple, Union from typing import Any, BinaryIO, Dict, Iterable, List, Optional, Tuple, Union
from urllib.error import HTTPError
from urllib.parse import urlencode from urllib.parse import urlencode
import json import json
@@ -318,7 +319,11 @@ class HubApi:
if MODELSCOPE_DOMAIN in os.environ: if MODELSCOPE_DOMAIN in os.environ:
endpoint = MODELSCOPE_URL_SCHEME + os.getenv(MODELSCOPE_DOMAIN) endpoint = MODELSCOPE_URL_SCHEME + os.getenv(MODELSCOPE_DOMAIN)
if not self.repo_exists(repo_id=repo_id, repo_type=repo_type, endpoint=endpoint): if not self.repo_exists(repo_id=repo_id, repo_type=repo_type, endpoint=endpoint):
raise NotExistError(f'Repo {repo_id} not exists on {endpoint}') raise HTTPError(url=endpoint + '/' + repo_id,
code=404,
msg=f'Repo {repo_id} not exists on {endpoint}',
hdrs={'Content-Type': 'text/html'},
fp=None)
else: else:
return endpoint return endpoint
@@ -330,7 +335,11 @@ class HubApi:
alternative_endpoint = get_endpoint(cn_site=(not check_cn_first)) alternative_endpoint = get_endpoint(cn_site=(not check_cn_first))
if not self.repo_exists( if not self.repo_exists(
repo_id, repo_type=repo_type, endpoint=alternative_endpoint): repo_id, repo_type=repo_type, endpoint=alternative_endpoint):
raise NotExistError(f'Repo {repo_id} not exists on either {prefer_endpoint} or {alternative_endpoint}') raise HTTPError(url=alternative_endpoint + '/' + repo_id,
code=404,
msg=f'Repo {repo_id} not exists on either {prefer_endpoint} or {alternative_endpoint}',
hdrs={'Content-Type': 'text/html'},
fp=None)
else: else:
return alternative_endpoint return alternative_endpoint
else: else: