From 003a84bc80a8024a9fd50c606485b9d9089a5dd4 Mon Sep 17 00:00:00 2001 From: "mulin.lyh" Date: Sat, 25 May 2024 12:40:19 +0800 Subject: [PATCH] fix ut --- tests/cli/test_download_cmd.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/cli/test_download_cmd.py b/tests/cli/test_download_cmd.py index 6059fa12..855ce9ff 100644 --- a/tests/cli/test_download_cmd.py +++ b/tests/cli/test_download_cmd.py @@ -53,12 +53,12 @@ class DownloadCMDTest(unittest.TestCase): super().tearDown() def test_download(self): - cmd = f'python -m modelscope.cli.cli download {self.model_id}' + cmd = f'python -m modelscope.cli.cli download --model {self.model_id}' stat, output = subprocess.getstatusoutput(cmd) self.assertEqual(stat, 0) def test_download_with_cache(self): - cmd = f'python -m modelscope.cli.cli download {self.model_id} --cache_dir {self.tmp_dir}' + cmd = f'python -m modelscope.cli.cli download --model {self.model_id} --cache_dir {self.tmp_dir}' stat, output = subprocess.getstatusoutput(cmd) if stat != 0: print(output) @@ -68,7 +68,7 @@ class DownloadCMDTest(unittest.TestCase): f'{self.tmp_dir}/{self.model_id}/{download_model_file_name}')) def test_download_with_revision(self): - cmd = f'python -m modelscope.cli.cli download {self.model_id} --revision {self.revision}' + cmd = f'python -m modelscope.cli.cli download --model {self.model_id} --revision {self.revision}' stat, output = subprocess.getstatusoutput(cmd) if stat != 0: print(output)