mirror of
https://github.com/modelscope/modelscope.git
synced 2026-02-24 12:10:09 +01:00
ok Merge branch 'master' of https://github.com/modelscope/modelscope into release/1.22
This commit is contained in:
@@ -742,13 +742,14 @@ class HubApi:
|
||||
|
||||
Args:
|
||||
repo_id (`str`): The repo id to use
|
||||
filename (`str`): The queried filename
|
||||
filename (`str`): The queried filename, if the file exists in a sub folder,
|
||||
please pass <sub-folder-name>/<file-name>
|
||||
revision (`Optional[str]`): The repo revision
|
||||
Returns:
|
||||
The query result in bool value
|
||||
"""
|
||||
files = self.get_model_files(repo_id, revision=revision)
|
||||
files = [file['Name'] for file in files]
|
||||
files = self.get_model_files(repo_id, recursive=True, revision=revision)
|
||||
files = [file['Path'] for file in files]
|
||||
return filename in files
|
||||
|
||||
def create_dataset(self,
|
||||
|
||||
41
tests/hub/test_file_exists.py
Normal file
41
tests/hub/test_file_exists.py
Normal file
@@ -0,0 +1,41 @@
|
||||
# Copyright (c) Alibaba, Inc. and its affiliates.
|
||||
import os
|
||||
import shutil
|
||||
import tempfile
|
||||
import unittest
|
||||
import uuid
|
||||
from os.path import expanduser
|
||||
|
||||
from requests import delete
|
||||
|
||||
from modelscope.hub.api import HubApi
|
||||
from modelscope.hub.constants import Licenses, ModelVisibility
|
||||
from modelscope.hub.errors import NotExistError
|
||||
from modelscope.hub.file_download import model_file_download
|
||||
from modelscope.hub.git import GitCommandWrapper
|
||||
from modelscope.hub.repository import Repository
|
||||
from modelscope.utils.constant import ModelFile
|
||||
from modelscope.utils.logger import get_logger
|
||||
from modelscope.utils.test_utils import (TEST_ACCESS_TOKEN1,
|
||||
TEST_MODEL_CHINESE_NAME,
|
||||
TEST_MODEL_ORG, delete_credential)
|
||||
|
||||
logger = get_logger()
|
||||
logger.setLevel('DEBUG')
|
||||
DEFAULT_GIT_PATH = 'git'
|
||||
download_model_file_name = 'test.bin'
|
||||
|
||||
|
||||
class FileExistsTest(unittest.TestCase):
|
||||
|
||||
def test_file_exsists(self):
|
||||
api = HubApi()
|
||||
self.assertTrue(
|
||||
api.file_exists('iic/gte_Qwen2-7B-instruct', 'added_tokens.json'))
|
||||
self.assertTrue(
|
||||
api.file_exists('iic/gte_Qwen2-7B-instruct',
|
||||
'1_Pooling/config.json'))
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user