Skip to content

Commit 09e847f

Browse files
author
marwan37
committed
add docker settings in pipeline definitions
1 parent 94d8e62 commit 09e847f

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

omni-reader/pipelines/batch_pipeline.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,7 @@
4242
)
4343

4444

45-
# settings={"docker": docker_settings}
46-
@pipeline()
45+
@pipeline(settings={"docker": docker_settings})
4746
def batch_ocr_pipeline(
4847
image_paths: Optional[List[str]] = None,
4948
image_folder: Optional[str] = None,

omni-reader/pipelines/evaluation_pipeline.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,8 +44,7 @@
4444
)
4545

4646

47-
# settings={"docker": docker_settings}
48-
@pipeline()
47+
@pipeline(settings={"docker": docker_settings})
4948
def ocr_evaluation_pipeline(
5049
ground_truth_folder: Optional[str] = None,
5150
ground_truth_files: Optional[List[str]] = None,

0 commit comments

Comments
 (0)