Let TF-GNN choose between keras or tf_keras consistently with TF 2.15:

tensorflow / gnn

TensorFlow GNN is a library to build Graph Neural Networks on the TensorFlow platform.

Apache License 2.0

1.3k stars 167 forks source link

Let TF-GNN choose between keras or tf_keras consistently with TF 2.15: #785

Closed copybara-service[bot] closed 4 months ago

copybara-service[bot] commented 4 months ago

Let TF-GNN choose between keras or tf_keras consistently with TF 2.15: both provide Keras 2.15, but it matters which one is used, because they have separate class hierarchies and global registries.

Along the way, refactor the nested case distinctions of tf_internal.py into a clear list of supported older TF/Keras versions.