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

PiperOrigin-RevId: 259362404

Author: copybara-github

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()

tensorflow_datasets/core/tfrecords_reader.py

@@ -22,7 +22,6 @@
 import functools
 import itertools
 import math
-import os
 import re
 
 import attr