From e87c70aef4f6ad7300434cc0e4064a174e371e78 Mon Sep 17 00:00:00 2001
From: geetachavan1 <53313357+geetachavan1@users.noreply.github.com>
Date: Wed, 3 Mar 2021 11:36:24 -0800
Subject: [PATCH] Update tpu_embedding.py

---
 tensorflow/python/tpu/tpu_embedding.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tensorflow/python/tpu/tpu_embedding.py b/tensorflow/python/tpu/tpu_embedding.py
index 783d98ec19e..67c1393696c 100644
--- a/tensorflow/python/tpu/tpu_embedding.py
+++ b/tensorflow/python/tpu/tpu_embedding.py
@@ -329,8 +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'])