diff --git a/tensorflow/python/estimator/canned/dnn_test.py b/tensorflow/python/estimator/canned/dnn_test.py index 2174977c397..c5af8e86876 100644 --- a/tensorflow/python/estimator/canned/dnn_test.py +++ b/tensorflow/python/estimator/canned/dnn_test.py @@ -707,6 +707,7 @@ class DNNClassifierIntegrationTest(test.TestCase): def tearDown(self): if self._model_dir: + writer_cache.FileWriterCache.clear() shutil.rmtree(self._model_dir) def _test_complete_flow(