Fix the import statements in neural_clustering, such that neural_clustering is imported as a standalone module outside NSL.

PiperOrigin-RevId: 353685620

Author: yueqiw

research/neural_clustering/__init__.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 """Neural Clustering."""
 
-from neural_structured_learning.research.neural_clustering import data_generators
-from neural_structured_learning.research.neural_clustering import models
-from neural_structured_learning.research.neural_clustering import utils
+from neural_clustering import data_generators
+from neural_clustering import models
+from neural_clustering import utils
 
 __all__ = ["data_generators", "models", "utils"]

research/neural_clustering/data_generators/__init__.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 """Data generators for neural clustering."""
 
-from neural_structured_learning.research.neural_clustering.data_generators.mog import MOGGenerator
-from neural_structured_learning.research.neural_clustering.data_generators.partition import CRPGenerator
+from neural_clustering.data_generators.mog import MOGGenerator
+from neural_clustering.data_generators.partition import CRPGenerator
 
 __all__ = ["CRPGenerator", "MOGGenerator"]

research/neural_clustering/data_generators/mog_test.py

@@ -15,7 +15,7 @@
 
 from absl.testing import absltest
 from absl.testing import parameterized
-from neural_structured_learning.research.neural_clustering.data_generators import mog
+from neural_clustering.data_generators import mog
 import numpy as np
 
 

research/neural_clustering/data_generators/partition_test.py

@@ -15,7 +15,7 @@
 
 from absl.testing import absltest
 from absl.testing import parameterized
-from neural_structured_learning.research.neural_clustering.data_generators import partition
+from neural_clustering.data_generators import partition
 import numpy as np
 
 

research/neural_clustering/models/__init__.py

@@ -13,11 +13,11 @@
 # limitations under the License.
 """Keras API for neural clustering models."""
 
-from neural_structured_learning.research.neural_clustering.models.ncp_base import NCPBase
-from neural_structured_learning.research.neural_clustering.models.ncp_models import NCPWithMLP
-from neural_structured_learning.research.neural_clustering.models.ncp_wrapper import CategoricalSampler
-from neural_structured_learning.research.neural_clustering.models.ncp_wrapper import GreedySampler
-from neural_structured_learning.research.neural_clustering.models.ncp_wrapper import NCPWrapper
+from neural_clustering.models.ncp_base import NCPBase
+from neural_clustering.models.ncp_models import NCPWithMLP
+from neural_clustering.models.ncp_wrapper import CategoricalSampler
+from neural_clustering.models.ncp_wrapper import GreedySampler
+from neural_clustering.models.ncp_wrapper import NCPWrapper
 
 __all__ = [
     'NCPBase', 'NCPWithMLP', 'NCPWrapper', 'GreedySampler', 'CategoricalSampler'

research/neural_clustering/models/ncp_models.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 """Neural Clustering Process models with concrete neural network layers."""
 
-from neural_structured_learning.research.neural_clustering.models.ncp_base import NCPBase
+from neural_clustering.models.ncp_base import NCPBase
 import tensorflow as tf
 
 

research/neural_clustering/models/ncp_test.py

@@ -14,7 +14,7 @@
 """Tests for neural_structured_learning.research.neural_clustering.models.ncp."""
 
 from absl.testing import parameterized
-from neural_structured_learning.research.neural_clustering import models
+from neural_clustering import models
 import numpy as np
 import tensorflow as tf
 

research/neural_clustering/models/ncp_wrapper.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 """NCP model wrapper and cluster samplers."""
 
-from neural_structured_learning.research.neural_clustering.utils.data_utils import batch_remap_label_ids
+from neural_clustering.utils.data_utils import batch_remap_label_ids
 import tensorflow as tf
 
 

research/neural_clustering/utils/__init__.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 """Utility functions for neural clustering."""
 
-from neural_structured_learning.research.neural_clustering.utils.data_utils import batch_remap_label_ids
-from neural_structured_learning.research.neural_clustering.utils.data_utils import remap_label_ids
+from neural_clustering.utils.data_utils import batch_remap_label_ids
+from neural_clustering.utils.data_utils import remap_label_ids
 
 __all__ = ["batch_remap_label_ids", "remap_label_ids"]

research/neural_clustering/utils/data_utils_test.py

@@ -15,7 +15,7 @@
 
 from absl.testing import absltest
 from absl.testing import parameterized
-from neural_structured_learning.research.neural_clustering.utils import data_utils
+from neural_clustering.utils import data_utils
 import numpy as np