fix comments

tastelikefeet · Sep 14, 2024 · f805f9e · f805f9e
1 parent 9ee9b72
commit f805f9e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/modelscope/hub/file_download.py b/modelscope/hub/file_download.py
@@ -466,6 +466,7 @@ def http_get_model_file(
                 desc='Downloading [' + file_name + ']',
             )
             if file_size == 0:
+                # Avoid empty file server request
                 with open(temp_file_path, 'w'):
                     progress.update(1)
                     progress.close()

diff --git a/tests/hub/test_hub_empty_file.py b/tests/hub/test_hub_empty_file.py
@@ -17,7 +17,7 @@ def tearDown(self):
         shutil.rmtree(self.work_dir, ignore_errors=True)
 
     def test_download_empty_file(self):
-        model_dir = snapshot_download('AI-ModelScope/GroundingDINO', cache_dir=self.work_dir)
+        model_dir = snapshot_download('tastelikefeet/test_empty_download', cache_dir=self.work_dir)
         self.assertTrue(model_dir is not None)
         self.assertTrue(os.path.exists(os.path.join(model_dir, '1.txt')))
         self.assertTrue(os.path.exists(os.path.join(model_dir, 'configuration.json')))