xla

9952e386 - fix type annotation for 3.8 (#7701)