def _did_process(self, data_container: DataContainer,
context: ExecutionContext) -> DataContainer:
"""
Reaugment the flattened data container.
:param data_container: data container to then_unflatten
:param context: execution context
:return: data container
"""
data_container = BaseStep._did_process(self, data_container, context)
if self.then_unflatten:
data_container.set_data_inputs(
self._reaugment_list(data_container.data_inputs, self.len_di))
data_container.set_expected_outputs(
self._reaugment_list(data_container.expected_outputs,
self.len_eo))
self.len_di = []
self.len_eo = []
return data_container
def _did_process(self, data_container, context):
data_container = BaseStep._did_process(self, data_container, context)
return data_container.reduce_dim()