resolve data dep size when adding data to SampleBlockMeta

Author: FynnBe

bioimageio/core/sample.py

@@ -251,7 +251,13 @@ def get_member_halo(m: MemberId, round: Callable[[float], int]):
 
     def with_data(self, data: PerMember[Tensor], *, stat: Stat) -> SampleBlock:
         return SampleBlock(
-            sample_shape=self.sample_shape,
+            sample_shape={
+                m: {
+                    a: data[m].tagged_shape[a] if s == -1 else s
+                    for a, s in member_shape.items()
+                }
+                for m, member_shape in self.sample_shape.items()
+            },
             sample_id=self.sample_id,
             blocks={
                 m: Block.from_meta(b, data=data[m]) for m, b in self.blocks.items()