if isinstance(processed_inputs, LazyDict):
                processed_inputs = {
                    k: v for k, v in processed_inputs.data.items() if k not in processed_inputs.keys_to_format
                }
                returned_lazy_dict = True
            else:
                returned_lazy_dict = False
            if update_data is None:
                # Check if the function returns updated examples
                updatable_types = (Mapping, pa.Table, pd.DataFrame)
                if config.POLARS_AVAILABLE and "polars" in sys.modules:
                    import polars as pl