if c in inputs_to_merge:
                            del inputs_to_merge[c]
                        if processed_inputs is inputs and c in processed_inputs:
                            del processed_inputs[c]
                transformed_example = {**inputs_to_merge, **processed_inputs}
                if self.features:
                    for c in self.features.keys():
                        if c not in transformed_example:
                            transformed_example[c] = None
                    transformed_example = self.features.decode_example(transformed_example)
                current_idx += 1
                if self._state_dict:
                    self._state_dict["previous_state_example_idx"] += 1
                yield key, transformed_example