Allows TensorRepresentation and its source feature to share name.

Sharing the name makes working with simpler schemas less error-prone and ambiguous.

PiperOrigin-RevId: 478584256

Author: iindyk

RELEASE.md

@@ -9,6 +9,9 @@
     in the same order as the terms in documents.
 *   `schema_utils.schema_as_feature_spec` now supports struct features as a way
     to describe `tf.SequenceExample` data.
+*   TensorRepresentations in schema used for
+    `schema_utils.schema_as_feature_spec` can now share name with their source
+    features.
 
 ## Bug Fixes and Other Changes
 

tensorflow_transform/tf_metadata/schema_utils.py

@@ -319,15 +319,18 @@ def schema_as_feature_spec(
           schema_proto, TENSOR_REPRESENTATION_GROUP))
   if tensor_representations is not None:
     for name, tensor_representation in tensor_representations.items():
-      if name in feature_by_name:
-        raise ValueError(
-            'Ragged TensorRepresentation name "{}" conflicts with a different '
-            'feature in the same schema.'.format(name))
       (feature_spec[name], domains[name]) = (
           _ragged_tensor_representation_as_feature_spec(name,
                                                         tensor_representation,
                                                         feature_by_name,
                                                         string_domains))
+      # At this point `feature_by_name` does not have source features for this
+      # tensor representation. If there's still a feature with the same name,
+      # then it would result in a name conflict.
+      if name in feature_by_name:
+        raise ValueError(
+            'Ragged TensorRepresentation name "{}" conflicts with a different '
+            'feature in the same schema.'.format(name))
 
   # Generate a `tf.FixedLenFeature` or `tf.VarLenFeature` for each element of
   # `schema_proto.feature` that was not referenced by a `SparseFeature` or a

tensorflow_transform/tf_metadata/schema_utils_test_cases.py

@@ -785,6 +785,38 @@
                       row_splits_dtype=tf.int64),
           },
       },
+      {
+          'testcase_name':
+              'ragged_tensor_and_feature_same_name',
+          'ascii_proto':
+              """
+              feature {
+                name: "ragged"
+                type: FLOAT
+              }
+              tensor_representation_group {
+                key: ""
+                value {
+                  tensor_representation {
+                    key: "ragged"
+                    value {
+                      ragged_tensor {
+                        feature_path { step: "ragged" }
+                      }
+                    }
+                  }
+                }
+              }
+            """,
+          'feature_spec': {
+              'ragged':
+                  tf.io.RaggedFeature(
+                      tf.float32,
+                      value_key='ragged',
+                      partitions=[],
+                      row_splits_dtype=tf.int64),
+          },
+      },
   ])
 
   NON_ROUNDTRIP_SCHEMAS.extend([{