def METHOD_NAME(self, pa_table: pa.Table) -> pa.Table:
    if self.info.features is not None:
        # more expensive cast to support nested features with keys in a different order
        # allows str <-> int/float or str to Audio for example
        pa_table = table_cast(pa_table, self.info.features.arrow_schema)
    return pa_table