trax - fix neural_gpu imports

T2T Team · copybara-github · commit 691abb198806 · 2019-05-07T16:33:38.000-07:00
PiperOrigin-RevId: 247107420
diff --git a/tensor2tensor/trax/models/neural_gpu.py b/tensor2tensor/trax/models/neural_gpu.py
@@ -17,7 +17,6 @@
 
 from __future__ import absolute_import
 from __future__ import division
-from __future__ import google_type_annotations
 from __future__ import print_function
 
 from tensor2tensor.trax import layers as tl
