From dc24408f8fa2c78222d4de0a704f3a73dc074f1d Mon Sep 17 00:00:00 2001 From: Yingda Chen Date: Wed, 22 Jan 2025 13:41:28 +0800 Subject: [PATCH] fix test --- tests/hub/test_hub_operation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/hub/test_hub_operation.py b/tests/hub/test_hub_operation.py index 252dce81..41589aed 100644 --- a/tests/hub/test_hub_operation.py +++ b/tests/hub/test_hub_operation.py @@ -155,9 +155,9 @@ class HubOperationTest(unittest.TestCase): self.prepare_case() snapshot_download_path = snapshot_download( model_id=self.model_id, revision=self.revision) - assert os.path.exists(snapshot_download_path) - assert '/hub/' in snapshot_download_path print(snapshot_download_path) + assert os.path.exists(snapshot_download_path) + assert 'models' in snapshot_download_path shutil.rmtree(snapshot_download_path) # download with cache_dir cache_dir = '/tmp/snapshot_download_cache_test'