diff --git a/timely/src/dataflow/operators/generic/builder_rc.rs b/timely/src/dataflow/operators/generic/builder_rc.rs index bca5f908c..98677b5f9 100644 --- a/timely/src/dataflow/operators/generic/builder_rc.rs +++ b/timely/src/dataflow/operators/generic/builder_rc.rs @@ -19,6 +19,7 @@ use crate::dataflow::channels::pullers::Counter as PullCounter; use crate::dataflow::operators::capability::Capability; use crate::dataflow::operators::generic::handles::{InputHandle, new_input_handle, OutputWrapper}; use crate::dataflow::operators::generic::operator_info::OperatorInfo; +use crate::dataflow::operators::generic::builder_raw::OperatorShape; use crate::logging::TimelyLogger as Logger; @@ -193,6 +194,11 @@ impl OperatorBuilder { self.builder.global() } + /// Return a reference to the operator's shape + pub fn shape(&self) -> &OperatorShape { + self.builder.shape() + } + /// Creates operator info for the operator. pub fn operator_info(&self) -> OperatorInfo { self.builder.operator_info()