Make sure all writer caches are closed before deleting directories in dnn_test.

2017-06-02 14:51:16 -07:00 · 2017-06-02 14:51:16 -07:00 · 05a6a13f79
commit 05a6a13f79
parent d0e761f8d5
1 changed files with 1 additions and 0 deletions
--- 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(