Fix import ordering.

PiperOrigin-RevId: 595063851
google-deepmind · Jan 2, 2024 · 03ab185 · 03ab185
1 parent 12a7c79
commit 03ab185
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sonnet/__init__.py b/sonnet/__init__.py
@@ -71,10 +71,10 @@
 from sonnet.src.recurrent import UnrolledLSTM
 from sonnet.src.recurrent import UnrolledRNN
 from sonnet.src.recurrent import VanillaRNN
-from sonnet.src.reshape import flatten
 from sonnet.src.reshape import Flatten
-from sonnet.src.reshape import reshape
+from sonnet.src.reshape import flatten
 from sonnet.src.reshape import Reshape
+from sonnet.src.reshape import reshape
 from sonnet.src.scale_gradient import scale_gradient
 from sonnet.src.sequential import Sequential
 from sonnet.src.utils import format_variables