Internal change

PiperOrigin-RevId: 463420241
tensorflow · Jul 26, 2022 · 417c8f0 · 417c8f0
1 parent 789af19
commit 417c8f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorboard_plugin/tensorboard_plugin_fairness_indicators/plugin_test.py b/tensorboard_plugin/tensorboard_plugin_fairness_indicators/plugin_test.py
@@ -69,7 +69,7 @@ def setUp(self):
     self._plugin = plugin.FairnessIndicatorsPlugin(self._context)
     self._multiplexer.Reload()
     wsgi_app = application.TensorBoardWSGI([self._plugin])
-    self._server = werkzeug_test.Client(wsgi_app, wrappers.BaseResponse)
+    self._server = werkzeug_test.Client(wsgi_app, wrappers.Response)
     self._routes = self._plugin.get_plugin_apps()
 
   def tearDown(self):