Internal change
Internal change

PiperOrigin-RevId: 446719663

google/yggdrasil-decision-forests,google/yggdrasil-decision-forests,google/yggdrasil-decision-forests,google/yggdrasil-decision-forests,google/yggdrasil-decision-forests,google/yggdrasil-decision-forests,google/yggdrasil-decision-forests