From 6c48771d3adce91b3cb33ba5120a84ed542166f0 Mon Sep 17 00:00:00 2001 From: zhijianma Date: Tue, 14 Nov 2023 13:41:00 +0800 Subject: [PATCH] change num_proc to 1 in test --- tests/ops/filter/test_clip_similarity_filter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/ops/filter/test_clip_similarity_filter.py b/tests/ops/filter/test_clip_similarity_filter.py index 0e33bb6ba..12fe318e1 100644 --- a/tests/ops/filter/test_clip_similarity_filter.py +++ b/tests/ops/filter/test_clip_similarity_filter.py @@ -15,8 +15,8 @@ class ClipSimilarityFilterTest(unittest.TestCase): cat_path = os.path.join(data_path, 'cat.jpg') img3_path = os.path.join(data_path, 'img3.jpg') - #hf_clip = 'openai/clip-vit-base-patch32' - hf_clip = '/Users/mazhijian/Documents/Project_2023/P01_LLM/C04_Data/clip-vit-base-patch32/' + hf_clip = 'openai/clip-vit-base-patch32' + def _run_filter(self, dataset: Dataset, target_list, op, num_proc=1): if Fields.stats not in dataset.features: