You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Switch tensorflow_gnn/experimental/sampler/eval_dag.py to tf_internal.py
for dealing with internals of different Keras versions.
PiperOrigin-RevId: 604655724
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.
PiperOrigin-RevId: 604619976