Skip to content

Commit 20fc9de

Browse files
committed
remove test code
1 parent 3c10162 commit 20fc9de

File tree

2 files changed

+0
-6
lines changed

2 files changed

+0
-6
lines changed

data_juicer/ops/base_op.py

-1
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,6 @@ def __init__(self, *args, **kwargs):
143143
self.history_key = kwargs.get('history_key', 'history')
144144

145145
self.batch_size = kwargs.get('batch_size', 1000)
146-
self.concurrency = kwargs.get('concurrency', 1)
147146

148147
# whether the model can be accelerated using cuda
149148
_accelerator = kwargs.get('accelerator', None)

data_juicer/ops/mapper/image_captioning_mapper.py

-5
Original file line numberDiff line numberDiff line change
@@ -283,11 +283,6 @@ def process_batched(self, samples, rank=None):
283283
:param samples:
284284
:return:
285285
"""
286-
import pyarrow as pa
287-
288-
if isinstance(samples, pa.Table):
289-
samples = samples.to_pydict()
290-
291286
# reconstruct samples from "dict of lists" to "list of dicts"
292287
reconstructed_samples = []
293288
for i in range(len(samples[self.text_key])):

0 commit comments

Comments
 (0)