Using the proper pipeline object for telemetry.

PiperOrigin-RevId: 441318280
tensorflow · Apr 12, 2022 · 2ac89ab · 2ac89ab
1 parent 0c3e662
commit 2ac89ab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensorflow_transform/beam/impl.py b/tensorflow_transform/beam/impl.py
@@ -1081,8 +1081,8 @@ def expand(self, dataset):
 
     # Gather telemetry on types of input features.
     _ = (
-        self.pipeline | 'CreateAnalyzeInputTensorRepresentations' >>
-        beam.Create([input_tensor_adapter_config.tensor_representations])
+        pipeline | 'CreateAnalyzeInputTensorRepresentations' >> beam.Create(
+            [input_tensor_adapter_config.tensor_representations])
         |
         'InstrumentAnalyzeInputTensors' >> telemetry.TrackTensorRepresentations(
             telemetry_util.AppendToNamespace(beam_common.METRICS_NAMESPACE,