Fix or ignore some pytype errors.#370

copybara-service[bot] wants to merge 1 commit intomaintensorflow/gnn:mainfrom

test_516590021tensorflow/gnn:test_516590021Copy head branch name to clipboard