n/a

zoyahav · tfx-copybara · commit fe2d383cf375 · 2022-08-23T09:03:16.000-07:00
PiperOrigin-RevId: 469473262
diff --git a/tensorflow_transform/analyzer_nodes.py b/tensorflow_transform/analyzer_nodes.py
@@ -35,9 +35,8 @@
 from tensorflow_transform import tf2_utils
 from tensorflow_transform import tf_utils
 from tensorflow_transform.graph_context import TFGraphContext
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 # pylint: disable=g-direct-tensorflow-import
diff --git a/tensorflow_transform/analyzers.py b/tensorflow_transform/analyzers.py
@@ -44,9 +44,8 @@
 from tensorflow_transform import schema_inference
 from tensorflow_transform import tf_utils
 from tfx_bsl import sketches
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 from typing_extensions import Literal
 
diff --git a/tensorflow_transform/beam/analysis_graph_builder.py b/tensorflow_transform/beam/analysis_graph_builder.py
@@ -26,9 +26,8 @@
 from tensorflow_transform.beam import analyzer_cache
 from tensorflow_transform.beam import beam_nodes
 from tensorflow_transform.beam import combiner_packing_util
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 
diff --git a/tensorflow_transform/beam/analysis_graph_builder_test.py b/tensorflow_transform/beam/analysis_graph_builder_test.py
@@ -23,9 +23,8 @@
 from tensorflow_transform import tf2_utils
 from tensorflow_transform.beam import analysis_graph_builder
 from tensorflow_transform import test_case
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 mock = tf.compat.v1.test.mock
diff --git a/tensorflow_transform/beam/analyzer_cache.py b/tensorflow_transform/beam/analyzer_cache.py
@@ -21,9 +21,8 @@
 
 import apache_beam as beam
 import tensorflow as tf
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 # This should be advanced whenever a non-backwards compatible change is made
diff --git a/tensorflow_transform/beam/beam_nodes.py b/tensorflow_transform/beam/beam_nodes.py
@@ -42,9 +42,8 @@
 
 import tensorflow as tf
 from tensorflow_transform import nodes
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 
diff --git a/tensorflow_transform/beam/cached_impl_test.py b/tensorflow_transform/beam/cached_impl_test.py
@@ -34,9 +34,8 @@
 from tensorflow_transform.beam import analyzer_cache
 from tensorflow_transform.beam import tft_unit
 from tensorflow_transform.tf_metadata import dataset_metadata
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 mock = tf.compat.v1.test.mock
diff --git a/tensorflow_transform/beam/combiner_packing_util.py b/tensorflow_transform/beam/combiner_packing_util.py
@@ -40,9 +40,8 @@
 from tensorflow_transform import analyzer_nodes
 from tensorflow_transform import nodes
 from tensorflow_transform.beam import beam_nodes
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 
diff --git a/tensorflow_transform/beam/common.py b/tensorflow_transform/beam/common.py
@@ -22,9 +22,8 @@
 from apache_beam.typehints import Union
 from tensorflow_transform import nodes
 from tfx_bsl.telemetry import util
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 NUMERIC_TYPE = Union[float, int]
diff --git a/tensorflow_transform/beam/context.py b/tensorflow_transform/beam/context.py
@@ -19,9 +19,8 @@
 
 import tensorflow as tf
 from tensorflow_transform import tf2_utils
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 
diff --git a/tensorflow_transform/beam/impl.py b/tensorflow_transform/beam/impl.py
@@ -86,9 +86,8 @@
 from tfx_bsl.tfxio import tf_example_record
 from tfx_bsl.tfxio.tensor_adapter import TensorAdapter
 from tfx_bsl.tfxio.tensor_adapter import TensorAdapterConfig
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 from tensorflow.python.framework import ops  # pylint: disable=g-direct-tensorflow-import
diff --git a/tensorflow_transform/beam/tft_beam_io/beam_metadata_io.py b/tensorflow_transform/beam/tft_beam_io/beam_metadata_io.py
@@ -25,9 +25,8 @@
 from tensorflow_transform import output_wrapper
 from tensorflow_transform.beam import common
 from tensorflow_transform.tf_metadata import metadata_io
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 
diff --git a/tensorflow_transform/experimental/analyzers.py b/tensorflow_transform/experimental/analyzers.py
@@ -37,9 +37,8 @@
 from tensorflow_transform import tf_utils
 from tfx_bsl import sketches
 
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 from typing_extensions import Protocol
 
diff --git a/tensorflow_transform/graph_context.py b/tensorflow_transform/graph_context.py
@@ -18,9 +18,8 @@
 from typing import Any, Dict, Optional
 
 import tensorflow as tf
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 
diff --git a/tensorflow_transform/graph_tools.py b/tensorflow_transform/graph_tools.py
@@ -36,9 +36,8 @@
 from tensorflow_transform import common_types
 from tensorflow_transform import nodes
 from tensorflow_transform import tf_utils
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 # pylint: disable=g-direct-tensorflow-import
 from tensorflow.python.framework import composite_tensor
diff --git a/tensorflow_transform/graph_tools_test.py b/tensorflow_transform/graph_tools_test.py
@@ -20,9 +20,8 @@
 import tensorflow as tf
 from tensorflow_transform import graph_tools
 from tensorflow_transform import test_case
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 # pylint: disable=g-direct-tensorflow-import
diff --git a/tensorflow_transform/mappers.py b/tensorflow_transform/mappers.py
@@ -62,9 +62,8 @@ def preprocessing_fn(inputs):
 from tensorflow_transform import gaussianization
 from tensorflow_transform import schema_inference
 from tensorflow_transform import tf_utils
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 
diff --git a/tensorflow_transform/nodes.py b/tensorflow_transform/nodes.py
@@ -29,9 +29,8 @@
 from typing import Collection, Optional, Tuple
 
 import pydot
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 
diff --git a/tensorflow_transform/nodes_test.py b/tensorflow_transform/nodes_test.py
@@ -16,9 +16,8 @@
 import tensorflow as tf
 from tensorflow_transform import nodes
 from tensorflow_transform import test_case
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 mock = tf.compat.v1.test.mock
diff --git a/tensorflow_transform/py_func/pyfunc_helper.py b/tensorflow_transform/py_func/pyfunc_helper.py
@@ -15,9 +15,8 @@
 
 import dill
 import tensorflow as tf
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 # pylint: disable=g-direct-tensorflow-import
diff --git a/tensorflow_transform/tf_metadata/schema_utils.py b/tensorflow_transform/tf_metadata/schema_utils.py
@@ -21,9 +21,8 @@
 from tensorflow_transform import common_types
 from tensorflow_transform.tf_metadata import schema_utils_legacy
 from tfx_bsl.tfxio import tensor_representation_util
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 from tensorflow_metadata.proto.v0 import path_pb2
diff --git a/tensorflow_transform/tf_utils.py b/tensorflow_transform/tf_utils.py
@@ -21,9 +21,8 @@
 import tensorflow as tf
 from tensorflow_transform import annotators
 from tensorflow_transform import common_types
-# TODO(https://issues.apache.org/jira/browse/SPARK-22674): Switch to
-# `collections.namedtuple` or `typing.NamedTuple` once the Spark issue is
-# resolved.
+# TODO(b/243513856): Switch to `collections.namedtuple` or `typing.NamedTuple`
+# once the Spark issue is resolved.
 from tfx_bsl.types import tfx_namedtuple
 
 # pylint: disable=g-direct-tensorflow-import
