Fix the integration test statisticsGen input key.

PiperOrigin-RevId: 360475343
speg03 · Mar 2, 2021 · 55e96f9 · 55e96f9
1 parent c25db3c
commit 55e96f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tfx/examples/chicago_taxi_pipeline/taxi_pipeline_runtime_parameter.py b/tfx/examples/chicago_taxi_pipeline/taxi_pipeline_runtime_parameter.py
@@ -110,7 +110,7 @@ def _create_parameterized_pipeline(
   # The input data location is parameterized by data_root
   example_gen = CsvExampleGen(input_base=data_root)
 
-  statistics_gen = StatisticsGen(input_data=example_gen.outputs['examples'])
+  statistics_gen = StatisticsGen(examples=example_gen.outputs['examples'])
   schema_gen = SchemaGen(
       statistics=statistics_gen.outputs['statistics'],
       infer_feature_shape=False)