Skip to content

Commit 8ee7d4a

Browse files
feat: add types to create_image_shard_downsample_tasks
1 parent b60164d commit 8ee7d4a

File tree

1 file changed

+13
-6
lines changed

1 file changed

+13
-6
lines changed

igneous/task_creation/image.py

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -622,16 +622,23 @@ def on_finish(self):
622622
return ImageShardTransferTaskIterator(bounds, shape)
623623

624624
def create_image_shard_downsample_tasks(
625-
cloudpath, mip=0, fill_missing=False,
626-
sparse=False, chunk_size=None,
627-
encoding=None, memory_target=MEMORY_TARGET,
628-
agglomerate=False, timestamp=None,
629-
factor=(2,2,1), bounds=None, bounds_mip=0,
625+
cloudpath:str,
626+
mip:int = 0,
627+
fill_missing:bool = False,
628+
sparse:bool = False,
629+
chunk_size:Optional[tuple[int,int,int]] = None,
630+
encoding:Optional[str] = None,
631+
memory_target:int = MEMORY_TARGET,
632+
agglomerate:bool = False,
633+
timestamp:Optional[int] = None,
634+
factor:tuple[int,int,int] = (2,2,1),
635+
bounds:Optional[Bbox] = None,
636+
bounds_mip:int = 0,
630637
encoding_level:Optional[int] = None,
631638
encoding_effort:Optional[int] = None,
632639
method=DownsampleMethods.AUTO,
633640
num_mips:Optional[int] = None,
634-
):
641+
) -> Iterator:
635642
"""
636643
Downsamples an existing image layer that may be
637644
sharded or unsharded to create a sharded layer.

0 commit comments

Comments
 (0)