Merge pull request tensorflow#796 from us:delete-unused-imports

PiperOrigin-RevId: 259362404
bdnoxt · Jul 22, 2019 · 0aba6e7 · 0aba6e7
2 parents 0cd5529 + f603447
commit 0aba6e7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tensorflow_datasets/core/dataset_builder_test.py b/tensorflow_datasets/core/dataset_builder_test.py
@@ -22,7 +22,6 @@
 import os
 
 from absl.testing import absltest
-from absl.testing import parameterized
 import numpy as np
 import tensorflow as tf
 from tensorflow_datasets import testing
@@ -390,9 +389,11 @@ def setUpClass(cls):
 
   @classmethod
   def tearDownClass(cls):
+    super(DatasetBuilderReadTest, cls).tearDownClass()
     testing.rm_tmp_dir(cls._tfds_tmp_dir)
 
   def setUp(self):
+    super(DatasetBuilderReadTest, self).setUp()
     self.builder = DummyDatasetSharedGenerator(data_dir=self._tfds_tmp_dir)
 
   @testing.run_in_graph_and_eager_modes()

diff --git a/tensorflow_datasets/core/tfrecords_reader.py b/tensorflow_datasets/core/tfrecords_reader.py
@@ -22,7 +22,6 @@
 import functools
 import itertools
 import math
-import os
 import re
 
 import attr