From 417c8f0433cee18998d0e60b1fabac0dd79af270 Mon Sep 17 00:00:00 2001 From: Nam Nguyen Date: Tue, 26 Jul 2022 13:55:08 -0700 Subject: [PATCH] Internal change PiperOrigin-RevId: 463420241 --- .../tensorboard_plugin_fairness_indicators/plugin_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorboard_plugin/tensorboard_plugin_fairness_indicators/plugin_test.py b/tensorboard_plugin/tensorboard_plugin_fairness_indicators/plugin_test.py index a136f2a7..1d245bc4 100644 --- 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):