Use self.get_temp_dir() instead of self.create_tempdir().

PiperOrigin-RevId: 539825411
google-parfait · Jun 13, 2023 · b520d3d · b520d3d
1 parent 1da584e
commit b520d3d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/dataset_grouper/core/integration_test.py b/dataset_grouper/core/integration_test.py
@@ -24,11 +24,12 @@
 class IntegrationTest(tf.test.TestCase):
 
   def test_pipeline_runs_and_data_loads(self):
-    data_dir = self.get_temp_dir()
+    temp_dir = self.get_temp_dir()
+    data_dir = os.path.join(temp_dir, 'data')
     dataset_builder = tfds.testing.DummyMnist(data_dir=data_dir)
     dataset_builder.download_and_prepare()
 
-    save_dir = self.create_tempdir()
+    save_dir = os.path.join(temp_dir, 'save')
     file_path_prefix = os.path.join(save_dir, 'mnist_test.tfrecord')
 
     mnist_pipeline = tfds_pipelines.tfds_to_tfrecords(