Skip to content

Commit

Permalink
Update direct import of TF to its correct location instead of importi…
Browse files Browse the repository at this point in the history 
…ng directly from the python init file.

This is in preparation for deleting imports from the python init file.

PiperOrigin-RevId: 544387630
  • Loading branch information
@fionalang @copybara-github
fionalang authored and copybara-github committed Jun 30, 2023
1 parent 26a685e commit 0c0c9c2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion sonnet/src/recurrent.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@

# pylint: disable=g-direct-tensorflow-import
# Required for specializing `UnrolledLSTM` per device.
from tensorflow.python import context as context_lib
from tensorflow.python.eager import context as context_lib
# pylint: enable=g-direct-tensorflow-import


Expand Down

0 comments on commit 0c0c9c2

Please sign in to comment.