pytorch

b0077074 - Add type annotation to get_worker_info