Skip to content

Commit 2ac89ab

Browse files
iindyktfx-copybara
authored andcommitted
Using the proper pipeline object for telemetry.
PiperOrigin-RevId: 441318280
1 parent 0c3e662 commit 2ac89ab

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Diff for: tensorflow_transform/beam/impl.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1081,8 +1081,8 @@ def expand(self, dataset):
10811081

10821082
# Gather telemetry on types of input features.
10831083
_ = (
1084-
self.pipeline | 'CreateAnalyzeInputTensorRepresentations' >>
1085-
beam.Create([input_tensor_adapter_config.tensor_representations])
1084+
pipeline | 'CreateAnalyzeInputTensorRepresentations' >> beam.Create(
1085+
[input_tensor_adapter_config.tensor_representations])
10861086
|
10871087
'InstrumentAnalyzeInputTensors' >> telemetry.TrackTensorRepresentations(
10881088
telemetry_util.AppendToNamespace(beam_common.METRICS_NAMESPACE,

0 commit comments

Comments
 (0)