Remove Python 2 compatibility code from optimizers package.

PiperOrigin-RevId: 297050950
google · Feb 25, 2020 · 47387a6 · 47387a6
1 parent 4914792
commit 47387a6
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 29 deletions.
diff --git a/trax/optimizers/adafactor.py b/trax/optimizers/adafactor.py
@@ -13,12 +13,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+# Lint as: python3
 """Adafactor optimizer class."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 from trax.math import numpy as np
 from trax.optimizers import base as opt_base
 

diff --git a/trax/optimizers/adam.py b/trax/optimizers/adam.py
@@ -13,12 +13,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+# Lint as: python3
 """Adam optimizer class."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 from trax.math import numpy as np
 from trax.optimizers import base as opt_base
 

diff --git a/trax/optimizers/base.py b/trax/optimizers/base.py
@@ -13,12 +13,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+# Lint as: python3
 """Trax base optimizer class."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 from trax.layers import base as layers
 from trax.math import numpy as np
 

diff --git a/trax/optimizers/momentum.py b/trax/optimizers/momentum.py
@@ -13,13 +13,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+# Lint as: python3
 """Nesterov momentum optimizer (also known as Nesterov Accelerated Gradient)."""
 
-
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 from trax.math import numpy as np
 from trax.optimizers import base
 

diff --git a/trax/optimizers/rms_prop.py b/trax/optimizers/rms_prop.py
@@ -13,12 +13,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+# Lint as: python3
 """RMSProp optimizer class."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 from trax.math import numpy as np
 from trax.optimizers import base as opt_base
 

diff --git a/trax/optimizers/sgd.py b/trax/optimizers/sgd.py
@@ -13,12 +13,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+# Lint as: python3
 """SGD optimizer class."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 from trax.optimizers import base as opt_base
 
 

diff --git a/trax/optimizers/sm3.py b/trax/optimizers/sm3.py
@@ -13,12 +13,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+# Lint as: python3
 """SM3 optimizer class."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 from trax.math import numpy as np
 from trax.optimizers import base as opt_base