xla

96fd3ce2 - Modify tpu version regex to match new names (#4557)

Modify tpu version regex to match new names (#4557) * Match litepod and lite version names * Formatting with yapf --------- Co-authored-by: root <root@t1v-n-804806aa-w-0.us-central2-b.c.tpu-pytorch.internal>