resolve data dep size when adding data to MetaBlock

Author: FynnBe

bioimageio/core/block.py

@@ -16,6 +16,7 @@
     Halo,
     HaloLike,
     PadMode,
+    SliceInfo,
     TotalNumberOfBlocks,
 )
 from .tensor import Tensor
@@ -34,6 +35,7 @@ def inner_data(self):
 
     def __post_init__(self):
         super().__post_init__()
+        assert not any(v == -1 for v in self.sample_shape.values()), self.sample_shape
         for a, s in self.data.sizes.items():
             slice_ = self.inner_slice[a]
             halo = self.halo.get(a, Halo(0, 0))
@@ -65,6 +67,27 @@ def get_transformed(
     ) -> Self:
         raise NotImplementedError
 
+    @classmethod
+    def from_meta(cls, meta: BlockMeta, data: Tensor) -> Self:
+        return cls(
+            sample_shape={
+                k: data.tagged_shape[k] if v == -1 else v
+                for k, v in meta.sample_shape.items()
+            },
+            inner_slice={
+                k: (
+                    SliceInfo(start=v.start, stop=data.tagged_shape[k])
+                    if v.stop == -1
+                    else v
+                )
+                for k, v in meta.inner_slice.items()
+            },
+            halo=meta.halo,
+            block_index=meta.block_index,
+            blocks_in_sample=meta.blocks_in_sample,
+            data=data,
+        )
+
 
 def split_tensor_into_blocks(
     tensor: Tensor,

bioimageio/core/sample.py

@@ -254,15 +254,7 @@ def with_data(self, data: PerMember[Tensor], *, stat: Stat) -> SampleBlock:
             sample_shape=self.sample_shape,
             sample_id=self.sample_id,
             blocks={
-                m: Block(
-                    sample_shape=self.sample_shape[m],
-                    inner_slice=b.inner_slice,
-                    halo=b.halo,
-                    block_index=b.block_index,
-                    blocks_in_sample=b.blocks_in_sample,
-                    data=data[m],
-                )
-                for m, b in self.blocks.items()
+                m: Block.from_meta(b, data=data[m]) for m, b in self.blocks.items()
             },
             stat=stat,
             block_index=self.block_index,