
"""TODO(waymo_open_dataset): Add a description here."""

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from tensorflow_datasets import testing
from tensorflow_datasets.object_detection import waymo_open_dataset


class WaymoOpenDatasetTest(testing.DatasetBuilderTestCase):
  DATASET_CLASS = waymo_open_dataset.WaymoOpenDataset
  SPLITS = {
      "train": 1,  # Number of fake train example
      "validation": 1,  # Number of fake test example
  }

  def setUp(self):
    super(WaymoOpenDatasetTest, self).setUp()
    self.builder._CLOUD_BUCKET = self.example_dir


if __name__ == "__main__":
  testing.test_main()


"""Test for waymo_open_dataset."""

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from tensorflow_datasets import testing
from tensorflow_datasets.object_detection import waymo_open_dataset


class WaymoOpenDatasetTest(testing.DatasetBuilderTestCase):
  DATASET_CLASS = waymo_open_dataset.WaymoOpenDataset
  SPLITS = {
      "train": 1,  # Number of fake train example
      "validation": 1,  # Number of fake test example
  }

  def setUp(self):
    super(WaymoOpenDatasetTest, self).setUp()
    self.builder._CLOUD_BUCKET = self.example_dir


if __name__ == "__main__":
  testing.test_main()

Add doc string for waymo open dataset
Add doc string for waymo open dataset
+ """Test for waymo_open_dataset."""
- 
- """TODO(waymo_open_dataset): Add a description here."""
  
  from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  
  from tensorflow_datasets import testing
  from tensorflow_datasets.object_detection import waymo_open_dataset
  
  
  class WaymoOpenDatasetTest(testing.DatasetBuilderTestCase):
    DATASET_CLASS = waymo_open_dataset.WaymoOpenDataset
    SPLITS = {
        "train": 1,  # Number of fake train example
        "validation": 1,  # Number of fake test example
    }
  
    def setUp(self):
      super(WaymoOpenDatasetTest, self).setUp()
      self.builder._CLOUD_BUCKET = self.example_dir
  
  
  if __name__ == "__main__":
    testing.test_main()
  
- 
Add doc string for waymo open dataset
## Code Before:

"""TODO(waymo_open_dataset): Add a description here."""

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from tensorflow_datasets import testing
from tensorflow_datasets.object_detection import waymo_open_dataset


class WaymoOpenDatasetTest(testing.DatasetBuilderTestCase):
  DATASET_CLASS = waymo_open_dataset.WaymoOpenDataset
  SPLITS = {
      "train": 1,  # Number of fake train example
      "validation": 1,  # Number of fake test example
  }

  def setUp(self):
    super(WaymoOpenDatasetTest, self).setUp()
    self.builder._CLOUD_BUCKET = self.example_dir


if __name__ == "__main__":
  testing.test_main()


## Instruction:
Add doc string for waymo open dataset
## Code After:
"""Test for waymo_open_dataset."""

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from tensorflow_datasets import testing
from tensorflow_datasets.object_detection import waymo_open_dataset


class WaymoOpenDatasetTest(testing.DatasetBuilderTestCase):
  DATASET_CLASS = waymo_open_dataset.WaymoOpenDataset
  SPLITS = {
      "train": 1,  # Number of fake train example
      "validation": 1,  # Number of fake test example
  }

  def setUp(self):
    super(WaymoOpenDatasetTest, self).setUp()
    self.builder._CLOUD_BUCKET = self.example_dir


if __name__ == "__main__":
  testing.test_main()
