Simplify github organization, so that all python modules are in `data…

…set_grouper/`.

PiperOrigin-RevId: 622967453

dataset_grouper/__init__.py

@@ -19,10 +19,10 @@
 __version__ = '0.2.1'
 
 
-from dataset_grouper.core import beam_transforms as beam
-from dataset_grouper.core import serialization
-from dataset_grouper.core.data_loaders import PartitionedDataset
-from dataset_grouper.core.tfds_pipelines import tfds_group_counts
-from dataset_grouper.core.tfds_pipelines import tfds_to_tfrecords
-from dataset_grouper.core.types import Example
-from dataset_grouper.core.types import GetKeyFn
+from dataset_grouper import beam_transforms as beam
+from dataset_grouper import serialization
+from dataset_grouper.data_loaders import PartitionedDataset
+from dataset_grouper.tfds_pipelines import tfds_group_counts
+from dataset_grouper.tfds_pipelines import tfds_to_tfrecords
+from dataset_grouper.types import Example
+from dataset_grouper.types import GetKeyFn

dataset_grouper/core/beam_transforms.py → dataset_grouper/beam_transforms.py

@@ -16,9 +16,9 @@
 from collections.abc import Iterable
 
 import apache_beam as beam
-from dataset_grouper.core import count_utils
-from dataset_grouper.core import serialization
-from dataset_grouper.core import types
+from dataset_grouper import count_utils
+from dataset_grouper import serialization
+from dataset_grouper import types
 import tensorflow as tf
 import tensorflow_datasets as tfds
 

dataset_grouper/core/beam_transforms_test.py → dataset_grouper/beam_transforms_test.py

@@ -17,7 +17,7 @@
 import apache_beam as beam
 from apache_beam.testing import test_pipeline
 from apache_beam.testing import util
-from dataset_grouper.core import beam_transforms
+from dataset_grouper import beam_transforms
 import numpy as np
 import tensorflow as tf
 import tensorflow_datasets as tfds

dataset_grouper/core/count_utils.py → dataset_grouper/count_utils.py

@@ -17,8 +17,8 @@
 from collections.abc import Iterable
 
 import apache_beam as beam
-from dataset_grouper.core import tf_utils
-from dataset_grouper.core import types
+from dataset_grouper import tf_utils
+from dataset_grouper import types
 
 
 GroupCount = collections.namedtuple(

dataset_grouper/core/count_utils_test.py → dataset_grouper/count_utils_test.py

@@ -17,7 +17,7 @@
 import apache_beam as beam
 from apache_beam.testing.util import assert_that
 from apache_beam.testing.util import equal_to
-from dataset_grouper.core import count_utils
+from dataset_grouper import count_utils
 
 
 class CountUtilsTest(absltest.TestCase):

dataset_grouper/core/data_loaders.py → dataset_grouper/data_loaders.py

@@ -15,7 +15,7 @@
 
 from typing import Optional, Union
 
-from dataset_grouper.core import serialization
+from dataset_grouper import serialization
 import tensorflow as tf
 import tensorflow_datasets as tfds
 

dataset_grouper/core/integration_test.py → dataset_grouper/integration_test.py

@@ -15,8 +15,8 @@
 import os.path
 
 import apache_beam as beam
-from dataset_grouper.core import data_loaders
-from dataset_grouper.core import tfds_pipelines
+from dataset_grouper import data_loaders
+from dataset_grouper import tfds_pipelines
 import tensorflow as tf
 import tensorflow_datasets as tfds
 

dataset_grouper/core/serialization.py → dataset_grouper/serialization.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 """Utilities for serializing examples."""
 
-from dataset_grouper.core import types
+from dataset_grouper import types
 import tensorflow as tf
 import tensorflow_datasets as tfds
 

dataset_grouper/core/serialization_test.py → dataset_grouper/serialization_test.py

@@ -14,7 +14,7 @@
 """Tests for serialization.py."""
 
 from absl.testing import parameterized
-from dataset_grouper.core import serialization
+from dataset_grouper import serialization
 import numpy as np
 import tensorflow as tf
 import tensorflow_datasets as tfds

dataset_grouper/core/tf_utils.py → dataset_grouper/tf_utils.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 """TensorFlow utils."""
 
-from dataset_grouper.core import types
+from dataset_grouper import types
 import tensorflow as tf
 
 

dataset_grouper/core/tf_utils_test.py → dataset_grouper/tf_utils_test.py

@@ -14,7 +14,7 @@
 """Tests for tf_utils."""
 
 from absl.testing import parameterized
-from dataset_grouper.core import tf_utils
+from dataset_grouper import tf_utils
 import tensorflow as tf
 
 

dataset_grouper/core/tfds_pipelines.py → dataset_grouper/tfds_pipelines.py

@@ -16,8 +16,8 @@
 from typing import Optional
 
 import apache_beam as beam
-from dataset_grouper.core import beam_transforms
-from dataset_grouper.core import types
+from dataset_grouper import beam_transforms
+from dataset_grouper import types
 import tensorflow as tf
 import tensorflow_datasets as tfds