Skip to content

Commit e9b91aa

Browse files
committed
fix nameclash
1 parent 6359d66 commit e9b91aa

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

bioimageio/core/prediction_pipeline/_combined_processing.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,14 +12,14 @@
1212

1313

1414
@dataclasses.dataclass
15-
class Processing:
15+
class ProcessingInfo:
1616
name: str
1717
kwargs: Dict[str, Any]
1818

1919

2020
@dataclasses.dataclass
2121
class TensorProcessingInfo:
22-
processing_steps: List[Processing]
22+
processing_steps: List[ProcessingInfo]
2323
assert_dtype_before: Optional[Union[str, Sequence[str]]] = None # throw AssertionError if data type doesn't match
2424
ensure_dtype_before: Optional[str] = None # cast data type if needed
2525
assert_dtype_after: Optional[Union[str, Sequence[str]]] = None # throw AssertionError if data type doesn't match
@@ -66,11 +66,11 @@ def from_tensor_specs(cls, tensor_specs: List[Union[nodes.InputTensor, nodes.Out
6666
# todo: assert nodes.InputTensor.dtype with assert_dtype_before?
6767
# todo: in the long run we do not want to limit model inputs to float32...
6868
combine_tensors[ts.name] = TensorProcessingInfo(
69-
[Processing(p.name, kwargs=p.kwargs) for p in ts.preprocessing], ensure_dtype_after="float32"
69+
[ProcessingInfo(p.name, kwargs=p.kwargs) for p in ts.preprocessing], ensure_dtype_after="float32"
7070
)
7171
elif isinstance(ts, nodes.OutputTensor):
7272
combine_tensors[ts.name] = TensorProcessingInfo(
73-
[Processing(p.name, kwargs=p.kwargs) for p in ts.postprocessing],
73+
[ProcessingInfo(p.name, kwargs=p.kwargs) for p in ts.postprocessing],
7474
ensure_dtype_after=ts.data_type,
7575
)
7676
else:

0 commit comments

Comments
 (0)