xla

9794392a - Specified vague "Any" type definitions (#4321)

Specified vague "Any" type definitions (#4321) * Specified vague "Any" type definitions * Changed function return type for `__getitem__()` * fixed formatting for yapf