From 20b3e15b74d8cbffc232f933e53c158ccc258357 Mon Sep 17 00:00:00 2001
From: geetachavan1 <53313357+geetachavan1@users.noreply.github.com>
Date: Wed, 3 Mar 2021 10:56:13 -0800
Subject: [PATCH] Update tpu_embedding.py

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

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'])