Skip to content

Commit 1cea292

Browse files
authored
add check (#116)
1 parent e19bdad commit 1cea292

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/data_designer/engine/dataset_builders/column_wise_builder.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,8 @@ def _run_cell_by_cell_generator(self, generator: ColumnGenerator) -> None:
171171
max_workers = MAX_CONCURRENCY_PER_NON_LLM_GENERATOR
172172
if isinstance(generator, WithLLMGeneration):
173173
max_workers = generator.inference_parameters.max_parallel_requests
174+
elif hasattr(generator.config, "max_parallel_requests"):
175+
max_workers = generator.config.max_parallel_requests
174176
self._fan_out_with_threads(generator, max_workers=max_workers)
175177

176178
def _run_full_column_generator(self, generator: ColumnGenerator) -> None:

0 commit comments

Comments
 (0)