Remove Python 3.6 workaround for kind retrieval
Remove Python 3.6 workaround for kind retrieval

PiperOrigin-RevId: 398716483
Change-Id: I7470df9221750db82bc3ad7e6914cd1dd830da28
GitOrigin-RevId: 2156aae8f63195e28ef1b965fdf3a4055daf1b6d

deepmind/xmanager,deepmind/xmanager