Skip to content

Commit 93f6667

Browse files
committed
fix assertion for no duplicate tensor names
1 parent 3de0d69 commit 93f6667

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

bioimageio/core/prediction_pipeline/_combined_processing.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,12 +73,12 @@ def __init__(self, inputs: List[nodes.InputTensor], outputs: List[nodes.OutputTe
7373
if any(self._req_output_stats[s] for s in SCOPES):
7474
raise NotImplementedError("computing statistics for output tensors not yet implemented")
7575

76-
assert not any(tn in self._req_output_stats for tn in self._req_input_stats)
77-
assert not any(tn in self._req_input_stats for tn in self._req_output_stats)
7876
self._computed_dataset_stats: Optional[Dict[str, Dict[Measure, Any]]] = None
7977

8078
self.input_tensor_names = [ipt.name for ipt in inputs]
8179
self.output_tensor_names = [out.name for out in outputs]
80+
assert not any(name in self.output_tensor_names for name in self.input_tensor_names)
81+
assert not any(name in self.input_tensor_names for name in self.output_tensor_names)
8282

8383
@property
8484
def required_input_dataset_statistics(self) -> Dict[str, Set[Measure]]:

0 commit comments

Comments
 (0)