diff --git a/tensorflow/python/tpu/tpu_embedding.py b/tensorflow/python/tpu/tpu_embedding.py
index 35ff13bed0a..783d98ec19e 100644
--- a/tensorflow/python/tpu/tpu_embedding.py
+++ b/tensorflow/python/tpu/tpu_embedding.py
@@ -329,7 +329,8 @@ ProximalAdagradSlotVariableName = collections.namedtuple(
 FtrlSlotVariableName = collections.namedtuple('FtrlSlotVariableName',
                                               ['accumulator', 'linear'])
 
-ProximalYogiSlotVariableNames = collections.namedtuple('ProximalYogiSlotVariableNames', ['v', 'm'])
+ProximalYogiSlotVariableNames = collections.namedtuple('ProximalYogiSlotVariableNames',
+                                                       ['v', 'm'])
 
 AdamSlotVariables = collections.namedtuple(
     'AdamSlotVariables', ['m', 'v'])