This is more future-proof because it is likely to simplify the upcoming changes to the sample pipelines. PiperOrigin-RevId: 511492014
