Update TODOs regarding CompositeTensor support.

PiperOrigin-RevId: 484299656

Author: zoyahav

tensorflow_transform/saved/saved_transform_io_v2.py

@@ -433,7 +433,7 @@ def finalize(self, input_tensor_keys: Iterable[str],
 
 
 # TODO(b/177606209): Remove once TF supports saving optimized functions.
-# TODO(b/169666856): WrappedFunction.prune does not support composite tensors.
+# TODO(b/129646028): WrappedFunction.prune does not support composite tensors.
 # Hence, add additional handling when supporting composite tensors in TFT.
 def optimize_concrete_function(
     concrete_function: function.ConcreteFunction,

tensorflow_transform/tf2_utils.py

@@ -170,7 +170,6 @@ def supply_missing_inputs(
         result[key] = tf.RaggedTensor.from_row_splits(
             result[key], row_splits, validate=False)
     else:
-      # TODO(b/169666856): Add support for generic CompositeTensors.
       raise ValueError('Received unsupported input tensor type. Only '
                        'dense/sparse/ragged tensors are currently supported.')
   return result