Skip to content

Commit

Permalink
change num_proc to 1 in test
Browse files Browse the repository at this point in the history
  • Loading branch information
zhijianma committed Nov 14, 2023
1 parent 7415502 commit 6c48771
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/ops/filter/test_clip_similarity_filter.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit 6c48771

Please sign in to comment.