wire up per stage params
This commit is contained in:
parent
89de05b7c5
commit
fb7a0996a8
|
@ -123,6 +123,11 @@ class ChainPipeline:
|
||||||
kwargs.keys(),
|
kwargs.keys(),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
per_stage_params = params
|
||||||
|
if "params" in kwargs:
|
||||||
|
per_stage_params = kwargs["params"]
|
||||||
|
kwargs.pop("params")
|
||||||
|
|
||||||
# the stage must be split and tiled if any image is larger than the selected/max tile size
|
# the stage must be split and tiled if any image is larger than the selected/max tile size
|
||||||
must_tile = any(
|
must_tile = any(
|
||||||
[
|
[
|
||||||
|
@ -159,7 +164,7 @@ class ChainPipeline:
|
||||||
worker,
|
worker,
|
||||||
server,
|
server,
|
||||||
stage_params,
|
stage_params,
|
||||||
kwargs["params"] if "params" in kwargs else params,
|
per_stage_params,
|
||||||
[source_tile],
|
[source_tile],
|
||||||
tile_mask=tile_mask,
|
tile_mask=tile_mask,
|
||||||
callback=callback,
|
callback=callback,
|
||||||
|
@ -201,7 +206,7 @@ class ChainPipeline:
|
||||||
worker,
|
worker,
|
||||||
server,
|
server,
|
||||||
stage_params,
|
stage_params,
|
||||||
params,
|
per_stage_params,
|
||||||
stage_sources,
|
stage_sources,
|
||||||
callback=callback,
|
callback=callback,
|
||||||
**kwargs,
|
**kwargs,
|
||||||
|
|
Loading…
Reference in New Issue