[r0.28.0 Cherry-pick] Fix the integration test statisticsGen input ke…

…y. (#3345) * Fix the integration test statisticsGen input key. PiperOrigin-RevId: 360475343 Co-authored-by: tfx-team <tensorflow-extended-nonhuman@googlegroups.com>
tensorflow · Mar 9, 2021 · feec708 · feec708
1 parent deb2f63
commit feec708
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)