tensorflow/gnn

examples/in_memory programs import functions incorrectly.

tbaker2 opened this issue · 1 comments

tbaker2 commented

In the examples/in_memory/tf_trainer.py file are the following lines (30-32).

from tensorflow_gnn.examples.in_memory import datasets
from tensorflow_gnn.examples.in_memory import models
from tensorflow_gnn.examples.in_memory import reader_utils

The tfgnn installed package doesn't have an examples directory so these lines fail. Removing the strings from tensorflow_gnn.examples.in_memory , and importing locally will run.

arnoegw commented

Hi @tbaker2 , thanks for reporting the issue! @mihirparadkar has fixed the import statements in commit
ea4231a (with some change to how they are exported out of Google's repo).