Merge pull request #387 from bioimage-io/optional_procs

skip simple procs if their input is not present
bioimage-io · May 2, 2024 · 5449618 · 5449618
2 parents 92e8f43 + 4706c95
commit 5449618
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/bioimageio/core/proc_ops.py b/bioimageio/core/proc_ops.py
@@ -73,6 +73,9 @@ def required_measures(self) -> Collection[Measure]:
     def get_output_shape(self, input_shape: PerAxis[int]) -> PerAxis[int]: ...
 
     def __call__(self, sample: Union[Sample, SampleBlock]) -> None:
+        if self.input not in sample.members:
+            return
+
         input_tensor = sample.members[self.input]
         output_tensor = self._apply(input_tensor, sample.stat)