From 03ab1852ea114271f4a6e295899de762de529bd9 Mon Sep 17 00:00:00 2001 From: Tom Hennigan Date: Tue, 2 Jan 2024 02:27:52 -0800 Subject: [PATCH] Fix import ordering. PiperOrigin-RevId: 595063851 --- sonnet/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sonnet/__init__.py b/sonnet/__init__.py index b2fbf49..714e844 100644 --- 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