Fix some dependencies#902

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

test_885616841tensorflow/gnn:test_885616841Copy head branch name to clipboard