Add infer_feature_shape=False back to the test pipeline for kubeflow end to end testing.

This is needed because input for `trip_seconds` is not always set.

PiperOrigin-RevId: 355282533

Co-authored-by: zhitaoli <zhitaoli@google.com>

Author: dhruvesh09

tfx/orchestration/kubeflow/v2/test_utils.py

@@ -131,7 +131,8 @@ def create_pipeline_components(
   statistics_gen = components.StatisticsGen(
       examples=example_gen.outputs['examples'])
   schema_gen = components.SchemaGen(
-      statistics=statistics_gen.outputs['statistics'])
+      statistics=statistics_gen.outputs['statistics'],
+      infer_feature_shape=False)
   example_validator = components.ExampleValidator(
       statistics=statistics_gen.outputs['statistics'],
       schema=schema_gen.outputs['schema'])

tfx/orchestration/kubeflow/v2/testdata/expected_full_taxi_pipeline_job.json

@@ -179,7 +179,7 @@
             "infer_feature_shape":{
               "runtimeValue":{
                 "constantValue":{
-                  "intValue":"1"
+                  "intValue":"0"
                 }
               }
             },