Path Lines of Code MANIFEST.in 12 conftest.py 6 package_build/ml-pipelines-sdk/package_config.py 1 package_build/tfx/package_config.py 1 package_config.py 1 setup.cfg 9 setup.py 205 tfx/__init__.py 2 tfx/benchmarks/__init__.py 1 tfx/benchmarks/benchmark_base.py 25 tfx/benchmarks/benchmark_dataset.py 37 tfx/benchmarks/benchmark_utils.py 20 tfx/benchmarks/datasets/__init__.py 1 tfx/benchmarks/datasets/chicago_taxi/__init__.py 1 tfx/benchmarks/datasets/chicago_taxi/dataset.py 190 tfx/benchmarks/regenerate_datasets.py 34 tfx/benchmarks/tfma_benchmark_base.py 194 tfx/benchmarks/tfma_benchmark_chicago_taxi.py 8 tfx/benchmarks/tfma_v2_benchmark_base.py 426 tfx/benchmarks/tfma_v2_benchmark_chicago_taxi.py 8 tfx/benchmarks/tft_benchmark_base.py 284 tfx/benchmarks/tft_benchmark_chicago_taxi.py 21 tfx/components/__init__.py 16 tfx/components/base/__init__.py 1 tfx/components/base/base_component.py 6 tfx/components/base/base_driver.py 6 tfx/components/base/base_executor.py 10 tfx/components/base/base_node.py 6 tfx/components/base/executor_spec.py 14 tfx/components/bulk_inferrer/__init__.py 1 tfx/components/bulk_inferrer/component.py 39 tfx/components/bulk_inferrer/executor.py 190 tfx/components/bulk_inferrer/prediction_to_example_utils.py 130 tfx/components/common_nodes/__init__.py 1 tfx/components/distribution_validator/__init__.py 1 tfx/components/distribution_validator/component.py 37 tfx/components/distribution_validator/executor.py 166 tfx/components/evaluator/__init__.py 1 tfx/components/evaluator/component.py 66 tfx/components/evaluator/constants.py 10 tfx/components/evaluator/executor.py 255 tfx/components/example_diff/__init__.py 1 tfx/components/example_diff/component.py 35 tfx/components/example_diff/executor.py 147 tfx/components/example_gen/__init__.py 1 tfx/components/example_gen/base_example_gen_executor.py 150 tfx/components/example_gen/component.py 80 tfx/components/example_gen/csv_example_gen/__init__.py 1 tfx/components/example_gen/csv_example_gen/component.py 25 tfx/components/example_gen/csv_example_gen/executor.py 137 tfx/components/example_gen/custom_executors/__init__.py 1 tfx/components/example_gen/custom_executors/avro_executor.py 23 tfx/components/example_gen/custom_executors/parquet_executor.py 24 tfx/components/example_gen/driver.py 128 tfx/components/example_gen/import_example_gen/__init__.py 1 tfx/components/example_gen/import_example_gen/component.py 25 tfx/components/example_gen/import_example_gen/executor.py 70 tfx/components/example_gen/input_processor.py 122 tfx/components/example_gen/utils.py 438 tfx/components/example_gen/write_split.py 60 tfx/components/example_validator/__init__.py 1 tfx/components/example_validator/component.py 30 tfx/components/example_validator/executor.py 81 tfx/components/example_validator/labels.py 7 tfx/components/experimental/__init__.py 1 tfx/components/experimental/data_view/__init__.py 1 tfx/components/experimental/data_view/binder_component.py 32 tfx/components/experimental/data_view/binder_executor.py 36 tfx/components/experimental/data_view/constants.py 2 tfx/components/experimental/data_view/provider_component.py 33 tfx/components/experimental/data_view/provider_executor.py 18 tfx/components/experimental/data_view/utils.py 10 tfx/components/infra_validator/__init__.py 1 tfx/components/infra_validator/component.py 29 tfx/components/infra_validator/error_types.py 10 tfx/components/infra_validator/executor.py 278 tfx/components/infra_validator/model_server_clients/__init__.py 1 tfx/components/infra_validator/model_server_clients/base_client.py 39 tfx/components/infra_validator/model_server_clients/tensorflow_serving_client.py 57 tfx/components/infra_validator/model_server_runners/__init__.py 1 tfx/components/infra_validator/model_server_runners/base_runner.py 16 tfx/components/infra_validator/model_server_runners/kubernetes_runner.py 266 tfx/components/infra_validator/model_server_runners/local_docker_runner.py 96 tfx/components/infra_validator/request_builder.py 194 tfx/components/infra_validator/serving_bins.py 117 tfx/components/infra_validator/types.py 15 tfx/components/model_validator/__init__.py 1 tfx/components/model_validator/component.py 23 tfx/components/model_validator/constants.py 14 tfx/components/model_validator/driver.py 39 tfx/components/model_validator/executor.py 133 tfx/components/pusher/__init__.py 1 tfx/components/pusher/component.py 48 tfx/components/pusher/executor.py 115 tfx/components/schema_gen/__init__.py 1 tfx/components/schema_gen/component.py 31 tfx/components/schema_gen/executor.py 47 tfx/components/schema_gen/import_schema_gen/component.py 14 tfx/components/schema_gen/import_schema_gen/executor.py 25 tfx/components/statistics_gen/__init__.py 1 tfx/components/statistics_gen/component.py 30 tfx/components/statistics_gen/executor.py 111 tfx/components/statistics_gen/stats_artifact_utils.py 16 tfx/components/trainer/__init__.py 1 tfx/components/trainer/component.py 75 tfx/components/trainer/constants.py 1 tfx/components/trainer/executor.py 113 tfx/components/trainer/fn_args_utils.py 126 tfx/components/trainer/rewriting/__init__.py 1 tfx/components/trainer/rewriting/converters.py 46 tfx/components/trainer/rewriting/rewriter.py 45 tfx/components/trainer/rewriting/rewriter_factory.py 38 tfx/components/trainer/rewriting/tfjs_rewriter.py 38 tfx/components/trainer/rewriting/tflite_rewriter.py 134 tfx/components/transform/__init__.py 1 tfx/components/transform/component.py 85 tfx/components/transform/executor.py 1115 tfx/components/transform/executor_utils.py 123 tfx/components/transform/labels.py 30 tfx/components/transform/messages.py 2 tfx/components/transform/run_executor.py 123 tfx/components/transform/stats_options_util.py 5 tfx/components/tuner/__init__.py 1 tfx/components/tuner/component.py 54 tfx/components/tuner/executor.py 74 tfx/components/util/__init__.py 1 tfx/components/util/examples_utils.py 50 tfx/components/util/model_utils.py 5 tfx/components/util/tfxio_utils.py 220 tfx/components/util/udf_utils.py 217 tfx/components/util/value_utils.py 30 tfx/dependencies.py 130 tfx/dsl/__init__.py 1 tfx/dsl/compiler/__init__.py 1 tfx/dsl/compiler/compiler.py 385 tfx/dsl/compiler/compiler_context.py 127 tfx/dsl/compiler/compiler_utils.py 196 tfx/dsl/compiler/constants.py 8 tfx/dsl/compiler/node_inputs_compiler.py 306 tfx/dsl/compiler/placeholder_utils.py 442 tfx/dsl/component/__init__.py 1 tfx/dsl/component/experimental/__init__.py 1 tfx/dsl/component/experimental/annotations.py 116 tfx/dsl/component/experimental/component_utils.py 53 tfx/dsl/component/experimental/container_component.py 50 tfx/dsl/component/experimental/decorators.py 272 tfx/dsl/component/experimental/executor_specs.py 91 tfx/dsl/component/experimental/function_parser.py 192 tfx/dsl/component/experimental/json_compat.py 71 tfx/dsl/component/experimental/placeholders.py 40 tfx/dsl/components/__init__.py 1 tfx/dsl/components/base/__init__.py 1 tfx/dsl/components/base/base_beam_component.py 19 tfx/dsl/components/base/base_beam_executor.py 81 tfx/dsl/components/base/base_component.py 121 tfx/dsl/components/base/base_driver.py 189 tfx/dsl/components/base/base_executor.py 80 tfx/dsl/components/base/base_node.py 138 tfx/dsl/components/base/executor_spec.py 98 tfx/dsl/components/common/__init__.py 1 tfx/dsl/components/common/importer.py 188 tfx/dsl/components/common/manual_node.py 18 tfx/dsl/components/common/resolver.py 151 tfx/dsl/context_managers/dsl_context.py 16 tfx/dsl/context_managers/dsl_context_manager.py 33 tfx/dsl/context_managers/dsl_context_registry.py 132 tfx/dsl/control_flow/for_each.py 31 tfx/dsl/control_flow/for_each_internal.py 32 tfx/dsl/experimental/__init__.py 1 tfx/dsl/experimental/conditionals/__init__.py 1 tfx/dsl/experimental/conditionals/conditional.py 33 tfx/dsl/experimental/latest_artifacts_resolver.py 6 tfx/dsl/experimental/latest_blessed_model_resolver.py 6 tfx/dsl/experimental/node_execution_options/__init__.py 1 tfx/dsl/experimental/node_execution_options/utils.py 11 tfx/dsl/experimental/spans_resolver.py 6 tfx/dsl/input_resolution/__init__.py 1 tfx/dsl/input_resolution/canned_resolver_functions.py 94 tfx/dsl/input_resolution/ops/all_spans_op.py 22 tfx/dsl/input_resolution/ops/consecutive_spans_op.py 49 tfx/dsl/input_resolution/ops/exclude_spans_op.py 18 tfx/dsl/input_resolution/ops/group_by_lineage_op.py 109 tfx/dsl/input_resolution/ops/latest_create_time_op.py 18 tfx/dsl/input_resolution/ops/latest_pipeline_run_outputs_op.py 66 tfx/dsl/input_resolution/ops/latest_policy_model_op.py 248 tfx/dsl/input_resolution/ops/latest_span_op.py 33 tfx/dsl/input_resolution/ops/latest_version_op.py 33 tfx/dsl/input_resolution/ops/ops.py 100 tfx/dsl/input_resolution/ops/ops_utils.py 117 tfx/dsl/input_resolution/ops/shuffle_op.py 12 tfx/dsl/input_resolution/ops/skip_if_empty_op.py 17 tfx/dsl/input_resolution/ops/skip_if_less_than_n_spans_op.py 23 tfx/dsl/input_resolution/ops/sliding_window_op.py 21 tfx/dsl/input_resolution/ops/span_driven_evaluator_inputs_op.py 92 tfx/dsl/input_resolution/ops/static_span_range_op.py 41 tfx/dsl/input_resolution/ops/training_range_op.py 65 tfx/dsl/input_resolution/ops/unnest_op.py 21 tfx/dsl/input_resolution/resolver_function.py 211 tfx/dsl/input_resolution/resolver_op.py 221 tfx/dsl/input_resolution/strategies/__init__.py 1 tfx/dsl/input_resolution/strategies/conditional_strategy.py 25 tfx/dsl/input_resolution/strategies/latest_artifact_strategy.py 26 tfx/dsl/input_resolution/strategies/latest_blessed_model_strategy.py 56 tfx/dsl/input_resolution/strategies/span_range_strategy.py 67 tfx/dsl/io/__init__.py 1 tfx/dsl/io/fileio.py 61 tfx/dsl/io/filesystem.py 52 tfx/dsl/io/filesystem_registry.py 52 tfx/dsl/io/plugins/__init__.py 1 tfx/dsl/io/plugins/local.py 89 tfx/dsl/io/plugins/tensorflow_gfile.py 85 tfx/dsl/placeholder/__init__.py 1 tfx/dsl/placeholder/placeholder.py 526 tfx/dsl/resolvers/__init__.py 1 tfx/experimental/__init__.py 1 tfx/experimental/distributed_inference/__init__.py 1 tfx/experimental/distributed_inference/graphdef_experiments/__init__.py 1 tfx/experimental/distributed_inference/graphdef_experiments/subgraph_partitioning/__init__.py 1 tfx/experimental/distributed_inference/graphdef_experiments/subgraph_partitioning/beam_pipeline.py 127 tfx/experimental/distributed_inference/graphdef_experiments/subgraph_partitioning/create_complex_graph.py 77 tfx/experimental/distributed_inference/graphdef_experiments/subgraph_partitioning/execution_spec.py 9 tfx/experimental/distributed_inference/graphdef_experiments/subgraph_partitioning/graph_partition.py 205 tfx/experimental/pipeline_testing/__init__.py 1 tfx/experimental/pipeline_testing/base_stub_component_launcher.py 31 tfx/experimental/pipeline_testing/base_stub_executor.py 39 tfx/experimental/pipeline_testing/executor_verifier_utils.py 168 tfx/experimental/pipeline_testing/pipeline_recorder.py 20 tfx/experimental/pipeline_testing/pipeline_recorder_utils.py 113 tfx/experimental/pipeline_testing/stub_component_launcher.py 4 tfx/experimental/templates/__init__.py 1 tfx/experimental/templates/container_based_test_case.py 241 tfx/experimental/templates/penguin/__init__.py 1 tfx/experimental/templates/penguin/e2e_tests/__init__.py 1 tfx/experimental/templates/penguin/kubeflow_runner.py 37 tfx/experimental/templates/penguin/local_runner.py 29 tfx/experimental/templates/penguin/models/__init__.py 1 tfx/experimental/templates/penguin/models/constants.py 6 tfx/experimental/templates/penguin/models/features.py 6 tfx/experimental/templates/penguin/models/model.py 134 tfx/experimental/templates/penguin/models/preprocessing.py 9 tfx/experimental/templates/penguin/pipeline/__init__.py 1 tfx/experimental/templates/penguin/pipeline/configs.py 17 tfx/experimental/templates/penguin/pipeline/pipeline.py 91 tfx/experimental/templates/taxi/__init__.py 1 tfx/experimental/templates/taxi/data_validation.ipynb 125 tfx/experimental/templates/taxi/e2e_tests/__init__.py 1 tfx/experimental/templates/taxi/kubeflow_runner.py 37 tfx/experimental/templates/taxi/kubeflow_v2_runner.py 30 tfx/experimental/templates/taxi/local_runner.py 29 tfx/experimental/templates/taxi/model_analysis.ipynb 105 tfx/experimental/templates/taxi/models/__init__.py 1 tfx/experimental/templates/taxi/models/estimator_model/__init__.py 1 tfx/experimental/templates/taxi/models/estimator_model/constants.py 3 tfx/experimental/templates/taxi/models/estimator_model/model.py 128 tfx/experimental/templates/taxi/models/features.py 17 tfx/experimental/templates/taxi/models/keras_model/__init__.py 1 tfx/experimental/templates/taxi/models/keras_model/constants.py 5 tfx/experimental/templates/taxi/models/keras_model/model.py 137 tfx/experimental/templates/taxi/models/preprocessing.py 37 tfx/experimental/templates/taxi/pipeline/__init__.py 1 tfx/experimental/templates/taxi/pipeline/configs.py 18 tfx/experimental/templates/taxi/pipeline/pipeline.py 110 tfx/extensions/__init__.py 1 tfx/extensions/experimental/__init__.py 1 tfx/extensions/experimental/kfp_compatibility/__init__.py 1 tfx/extensions/experimental/kfp_compatibility/kfp_container_component.py 59 tfx/extensions/experimental/kfp_compatibility/proto/kfp_component_spec.proto 80 tfx/extensions/google_cloud_ai_platform/__init__.py 1 tfx/extensions/google_cloud_ai_platform/bulk_inferrer/__init__.py 1 tfx/extensions/google_cloud_ai_platform/bulk_inferrer/component.py 66 tfx/extensions/google_cloud_ai_platform/bulk_inferrer/executor.py 179 tfx/extensions/google_cloud_ai_platform/cmle_runner.py 10 tfx/extensions/google_cloud_ai_platform/constants.py 21 tfx/extensions/google_cloud_ai_platform/prediction_clients.py 330 tfx/extensions/google_cloud_ai_platform/pusher/__init__.py 1 tfx/extensions/google_cloud_ai_platform/pusher/component.py 17 tfx/extensions/google_cloud_ai_platform/pusher/executor.py 96 tfx/extensions/google_cloud_ai_platform/runner.py 135 tfx/extensions/google_cloud_ai_platform/trainer/__init__.py 1 tfx/extensions/google_cloud_ai_platform/trainer/component.py 39 tfx/extensions/google_cloud_ai_platform/trainer/executor.py 74 tfx/extensions/google_cloud_ai_platform/training_clients.py 203 tfx/extensions/google_cloud_ai_platform/tuner/__init__.py 1 tfx/extensions/google_cloud_ai_platform/tuner/component.py 5 tfx/extensions/google_cloud_ai_platform/tuner/executor.py 200 tfx/extensions/google_cloud_big_query/__init__.py 1 tfx/extensions/google_cloud_big_query/example_gen/__init__.py 1 tfx/extensions/google_cloud_big_query/example_gen/component.py 26 tfx/extensions/google_cloud_big_query/example_gen/executor.py 29 tfx/extensions/google_cloud_big_query/experimental/__init__.py 1 tfx/extensions/google_cloud_big_query/experimental/elwc_example_gen/__init__.py 1 tfx/extensions/google_cloud_big_query/experimental/elwc_example_gen/component/__init__.py 1 tfx/extensions/google_cloud_big_query/experimental/elwc_example_gen/component/component.py 27 tfx/extensions/google_cloud_big_query/experimental/elwc_example_gen/component/executor.py 75 tfx/extensions/google_cloud_big_query/experimental/elwc_example_gen/proto/__init__.py 1 tfx/extensions/google_cloud_big_query/experimental/elwc_example_gen/proto/elwc_config.proto 22 tfx/extensions/google_cloud_big_query/pusher/__init__.py 1 tfx/extensions/google_cloud_big_query/pusher/component.py 17 tfx/extensions/google_cloud_big_query/pusher/executor.py 73 tfx/extensions/google_cloud_big_query/utils.py 49 tfx/orchestration/airflow/__init__.py 1 tfx/orchestration/airflow/airflow_component.py 60 tfx/orchestration/airflow/airflow_dag_runner.py 78 tfx/orchestration/airflow/airflow_runner.py 6 tfx/orchestration/beam/__init__.py 1 tfx/orchestration/beam/beam_dag_runner.py 187 tfx/orchestration/beam/legacy/__init__.py 1 tfx/orchestration/beam/legacy/beam_dag_runner.py 88 tfx/orchestration/config/__init__.py 1 tfx/orchestration/config/base_component_config.py 3 tfx/orchestration/config/config_utils.py 21 tfx/orchestration/config/docker_component_config.py 34 tfx/orchestration/config/kubernetes_component_config.py 9 tfx/orchestration/config/pipeline_config.py 30 tfx/orchestration/data_types.py 89 tfx/orchestration/data_types_utils.py 218 tfx/orchestration/experimental/__init__.py 1 tfx/orchestration/experimental/centralized_kubernetes_orchestrator/__init__.py 1 tfx/orchestration/experimental/centralized_kubernetes_orchestrator/entrypoint.py 6 tfx/orchestration/experimental/centralized_kubernetes_orchestrator/kubernetes_job_runner.py 153 tfx/orchestration/experimental/centralized_kubernetes_orchestrator/kubernetes_task_scheduler.py 83 tfx/orchestration/experimental/centralized_kubernetes_orchestrator/main.py 133 tfx/orchestration/experimental/centralized_kubernetes_orchestrator/service/__init__.py 1 tfx/orchestration/experimental/centralized_kubernetes_orchestrator/service/kubernetes_orchestrator_service.py 42 tfx/orchestration/experimental/centralized_kubernetes_orchestrator/service/proto/service.proto 48 tfx/orchestration/experimental/core/__init__.py 1 tfx/orchestration/experimental/core/async_pipeline_task_gen.py 319 tfx/orchestration/experimental/core/constants.py 9 tfx/orchestration/experimental/core/env.py 46 tfx/orchestration/experimental/core/event_observer.py 175 tfx/orchestration/experimental/core/garbage_collection.py 241 tfx/orchestration/experimental/core/mlmd_state.py 89 tfx/orchestration/experimental/core/orchestration_options.py 5 tfx/orchestration/experimental/core/pipeline_ops.py 1146 tfx/orchestration/experimental/core/pipeline_state.py 909 tfx/orchestration/experimental/core/post_execution_utils.py 170 tfx/orchestration/experimental/core/sample_mlmd_creator.py 105 tfx/orchestration/experimental/core/service_jobs.py 75 tfx/orchestration/experimental/core/sync_pipeline_task_gen.py 444 tfx/orchestration/experimental/core/task.py 112 tfx/orchestration/experimental/core/task_gen.py 8 tfx/orchestration/experimental/core/task_gen_utils.py 413 tfx/orchestration/experimental/core/task_manager.py 219 tfx/orchestration/experimental/core/task_queue.py 47 tfx/orchestration/experimental/core/task_scheduler.py 110 tfx/orchestration/experimental/core/task_schedulers/__init__.py 1 tfx/orchestration/experimental/core/task_schedulers/importer_task_scheduler.py 33 tfx/orchestration/experimental/core/task_schedulers/manual_task_scheduler.py 58 tfx/orchestration/experimental/core/task_schedulers/noop_task_scheduler.py 21 tfx/orchestration/experimental/core/task_schedulers/resolver_task_scheduler.py 12 tfx/orchestration/experimental/core/task_schedulers/subpipeline_task_scheduler.py 123 tfx/orchestration/experimental/core/testing/__init__.py 1 tfx/orchestration/experimental/interactive/__init__.py 1 tfx/orchestration/experimental/interactive/execution_result.py 18 tfx/orchestration/experimental/interactive/export_templates/__init__.py 1 tfx/orchestration/experimental/interactive/interactive_context.py 142 tfx/orchestration/experimental/interactive/notebook_extensions/__init__.py 1 tfx/orchestration/experimental/interactive/notebook_extensions/skip.py 11 tfx/orchestration/experimental/interactive/notebook_formatters.py 148 tfx/orchestration/experimental/interactive/notebook_utils.py 13 tfx/orchestration/experimental/interactive/standard_visualizations.py 60 tfx/orchestration/experimental/interactive/visualizations.py 26 tfx/orchestration/experimental/kubernetes/__init__.py 1 tfx/orchestration/experimental/kubernetes/container_entrypoint.py 60 tfx/orchestration/experimental/kubernetes/kubernetes_dag_runner.py 151 tfx/orchestration/experimental/kubernetes/kubernetes_remote_runner.py 163 tfx/orchestration/experimental/kubernetes/node_wrapper.py 26 tfx/orchestration/experimental/kubernetes/orchestrator_container_entrypoint.py 19 tfx/orchestration/experimental/kubernetes/yaml/jupyter.yaml 19 tfx/orchestration/experimental/kubernetes/yaml/kustomization.yaml 6 tfx/orchestration/experimental/kubernetes/yaml/mysql-pv.yaml 10 tfx/orchestration/experimental/kubernetes/yaml/mysql.yaml 42 tfx/orchestration/experimental/kubernetes/yaml/roles.yaml 15 tfx/orchestration/experimental/kubernetes/yaml/service-account.yaml 5 tfx/orchestration/kubeflow/__init__.py 1 tfx/orchestration/kubeflow/base_component.py 88 tfx/orchestration/kubeflow/container_entrypoint.py 294 tfx/orchestration/kubeflow/decorators.py 7 tfx/orchestration/kubeflow/kubeflow_dag_runner.py 224 tfx/orchestration/kubeflow/proto/kubeflow.proto 46 tfx/orchestration/kubeflow/runner.py 10 tfx/orchestration/kubeflow/utils.py 18 tfx/orchestration/kubeflow/v2/__init__.py 1 tfx/orchestration/kubeflow/v2/artifact_types/Artifact.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/Boolean.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/Dataset.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/ExampleAnomalies.yaml 9 tfx/orchestration/kubeflow/v2/artifact_types/ExampleStatistics.yaml 9 tfx/orchestration/kubeflow/v2/artifact_types/Examples.yaml 12 tfx/orchestration/kubeflow/v2/artifact_types/ExamplesDiff.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/File.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/Float.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/HyperParameters.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/InferenceResult.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/InfraBlessing.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/Integer.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/JsonValue.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/Metrics.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/Model.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/ModelBlessing.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/ModelEvaluation.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/ModelRun.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/PushedModel.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/Schema.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/Statistics.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/String.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/TransformCache.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/TransformGraph.yaml 2 tfx/orchestration/kubeflow/v2/artifact_types/TunerResults.yaml 2 tfx/orchestration/kubeflow/v2/compiler_utils.py 296 tfx/orchestration/kubeflow/v2/components/__init__.py 1 tfx/orchestration/kubeflow/v2/components/experimental/__init__.py 3 tfx/orchestration/kubeflow/v2/components/experimental/ai_platform_training_component.py 92 tfx/orchestration/kubeflow/v2/components/experimental/ai_platform_training_executor.py 35 tfx/orchestration/kubeflow/v2/container/__init__.py 1 tfx/orchestration/kubeflow/v2/container/kubeflow_v2_entrypoint_utils.py 173 tfx/orchestration/kubeflow/v2/container/kubeflow_v2_run_executor.py 113 tfx/orchestration/kubeflow/v2/e2e_tests/base_test_case.py 90 tfx/orchestration/kubeflow/v2/file_based_example_gen/__init__.py 1 tfx/orchestration/kubeflow/v2/file_based_example_gen/driver.py 94 tfx/orchestration/kubeflow/v2/kubeflow_v2_dag_runner.py 93 tfx/orchestration/kubeflow/v2/parameter_utils.py 18 tfx/orchestration/kubeflow/v2/pipeline_builder.py 117 tfx/orchestration/kubeflow/v2/step_builder.py 511 tfx/orchestration/kubeflow/v2/vertex_client_utils.py 25 tfx/orchestration/launcher/__init__.py 1 tfx/orchestration/launcher/base_component_launcher.py 124 tfx/orchestration/launcher/container_common.py 91 tfx/orchestration/launcher/docker_component_launcher.py 53 tfx/orchestration/launcher/in_process_component_launcher.py 38 tfx/orchestration/launcher/kubernetes_component_launcher.py 142 tfx/orchestration/local/__init__.py 1 tfx/orchestration/local/legacy/__init__.py 1 tfx/orchestration/local/legacy/local_dag_runner.py 49 tfx/orchestration/local/local_dag_runner.py 67 tfx/orchestration/local/runner_utils.py 88 tfx/orchestration/metadata.py 714 tfx/orchestration/metadata_test_utils.py 515 tfx/orchestration/mlmd_connection_manager.py 83 tfx/orchestration/node_proto_view.py 168 tfx/orchestration/pipeline.py 214 tfx/orchestration/portable/__init__.py 1 tfx/orchestration/portable/base_driver.py 12 tfx/orchestration/portable/base_driver_operator.py 20 tfx/orchestration/portable/base_executor_operator.py 33 tfx/orchestration/portable/beam_executor_operator.py 58 tfx/orchestration/portable/cache_utils.py 96 tfx/orchestration/portable/data_types.py 57 tfx/orchestration/portable/docker_executor_operator.py 88 tfx/orchestration/portable/execution_publish_utils.py 102 tfx/orchestration/portable/execution_watcher.py 70 tfx/orchestration/portable/importer_node_handler.py 78 tfx/orchestration/portable/input_resolution/channel_resolver.py 115 tfx/orchestration/portable/input_resolution/exceptions.py 17 tfx/orchestration/portable/input_resolution/input_graph_resolver.py 157 tfx/orchestration/portable/input_resolution/node_inputs_resolver.py 251 tfx/orchestration/portable/input_resolution/partition_utils.py 46 tfx/orchestration/portable/inputs_utils.py 87 tfx/orchestration/portable/kubernetes_executor_operator.py 141 tfx/orchestration/portable/launcher.py 435 tfx/orchestration/portable/merge_utils.py 93 tfx/orchestration/portable/mlmd/__init__.py 1 tfx/orchestration/portable/mlmd/artifact_lib.py 36 tfx/orchestration/portable/mlmd/common_utils.py 41 tfx/orchestration/portable/mlmd/context_lib.py 110 tfx/orchestration/portable/mlmd/event_lib.py 136 tfx/orchestration/portable/mlmd/execution_lib.py 336 tfx/orchestration/portable/mlmd/filter_query_builder.py 30 tfx/orchestration/portable/mlmd/store_ext.py 67 tfx/orchestration/portable/outputs_utils.py 241 tfx/orchestration/portable/partial_run_utils.py 414 tfx/orchestration/portable/python_driver_operator.py 25 tfx/orchestration/portable/python_executor_operator.py 65 tfx/orchestration/portable/resolver_node_handler.py 100 tfx/orchestration/portable/runtime_parameter_utils.py 97 tfx/orchestration/portable/system_node_handler.py 14 tfx/orchestration/portable/tfx_runner.py 64 tfx/orchestration/publisher.py 21 tfx/orchestration/python_execution_binary/__init__.py 1 tfx/orchestration/python_execution_binary/entrypoint.py 92 tfx/orchestration/python_execution_binary/python_execution_binary_utils.py 48 tfx/orchestration/tfx_runner.py 14 tfx/proto/bulk_inferrer.proto 70 tfx/proto/distribution_validator.proto 34 tfx/proto/evaluator.proto 44 tfx/proto/example_diff.proto 46 tfx/proto/example_gen.proto 183 tfx/proto/infra_validator.proto 241 tfx/proto/orchestration/__init__.py 1 tfx/proto/orchestration/driver_output.proto 27 tfx/proto/orchestration/executable_spec.proto 45 tfx/proto/orchestration/execution_invocation.proto 51 tfx/proto/orchestration/execution_result.proto 48 tfx/proto/orchestration/execution_watcher.proto 28 tfx/proto/orchestration/garbage_collection_policy.proto 52 tfx/proto/orchestration/local_deployment_config.proto 49 tfx/proto/orchestration/metadata.proto 31 tfx/proto/orchestration/pipeline.proto 710 tfx/proto/orchestration/placeholder.proto 236 tfx/proto/orchestration/platform_config.proto 53 tfx/proto/orchestration/run_state.proto 50 tfx/proto/pusher.proto 46 tfx/proto/range_config.proto 51 tfx/proto/trainer.proto 31 tfx/proto/transform.proto 24 tfx/proto/tuner.proto 25 tfx/scripts/__init__.py 1 tfx/scripts/run_component.py 102 tfx/scripts/run_executor.py 94 tfx/tfx.bzl 29 tfx/tools/__init__.py 1 tfx/tools/build_docs.py 57 tfx/tools/cli/__init__.py 1 tfx/tools/cli/cli_context.py 5 tfx/tools/cli/cli_main.py 12 tfx/tools/cli/commands/__init__.py 1 tfx/tools/cli/commands/pipeline.py 280 tfx/tools/cli/commands/run.py 253 tfx/tools/cli/commands/template.py 43 tfx/tools/cli/container_builder/__init__.py 1 tfx/tools/cli/container_builder/builder.py 48 tfx/tools/cli/container_builder/dockerfile.py 40 tfx/tools/cli/container_builder/labels.py 5 tfx/tools/cli/handler/__init__.py 1 tfx/tools/cli/handler/airflow_dag_runner_patcher.py 7 tfx/tools/cli/handler/airflow_handler.py 111 tfx/tools/cli/handler/base_handler.py 187 tfx/tools/cli/handler/beam_dag_runner_patcher.py 9 tfx/tools/cli/handler/beam_handler.py 77 tfx/tools/cli/handler/dag_runner_patcher.py 70 tfx/tools/cli/handler/handler_factory.py 61 tfx/tools/cli/handler/kubeflow_dag_runner_patcher.py 43 tfx/tools/cli/handler/kubeflow_handler.py 182 tfx/tools/cli/handler/kubeflow_v2_dag_runner_patcher.py 33 tfx/tools/cli/handler/local_dag_runner_patcher.py 9 tfx/tools/cli/handler/local_handler.py 6 tfx/tools/cli/handler/template_handler.py 111 tfx/tools/cli/handler/vertex_handler.py 106 tfx/tools/cli/labels.py 26 tfx/tools/cli/pip_utils.py 13 tfx/types/__init__.py 8 tfx/types/artifact.py 539 tfx/types/artifact_property.py 25 tfx/types/artifact_utils.py 183 tfx/types/channel.py 391 tfx/types/channel_utils.py 103 tfx/types/component_spec.py 271 tfx/types/experimental/__init__.py 1 tfx/types/experimental/simple_artifacts.py 9 tfx/types/node_common.py 32 tfx/types/resolved_channel.py 58 tfx/types/standard_artifacts.py 157 tfx/types/standard_component_specs.py 445 tfx/types/system_artifacts.py 12 tfx/types/system_executions.py 14 tfx/types/value_artifact.py 110 tfx/utils/__init__.py 1 tfx/utils/abc_utils.py 4 tfx/utils/channel.py 20 tfx/utils/dependency_utils.py 76 tfx/utils/deprecation_utils.py 94 tfx/utils/doc_controls.py 24 tfx/utils/docker_utils.py 15 tfx/utils/golden_utils.py 2 tfx/utils/governance_utils.py 7 tfx/utils/import_utils.py 100 tfx/utils/io_utils.py 140 tfx/utils/json_utils.py 110 tfx/utils/kube_utils.py 157 tfx/utils/logging_utils.py 39 tfx/utils/model_paths/__init__.py 1 tfx/utils/model_paths/tf_serving_flavor.py 21 tfx/utils/name_utils.py 49 tfx/utils/path_constants.py 9 tfx/utils/path_utils.py 62 tfx/utils/proto_utils.py 69 tfx/utils/retry.py 29 tfx/utils/status.py 35 tfx/utils/telemetry_utils.py 60 tfx/utils/topsort.py 55 tfx/utils/typing_utils.py 102 tfx/utils/version_utils.py 13 tfx/utils/writer_utils.py 9 tfx/v1/__init__.py 12 tfx/v1/components/__init__.py 17 tfx/v1/dsl/__init__.py 11 tfx/v1/dsl/components/__init__.py 6 tfx/v1/dsl/experimental/__init__.py 6 tfx/v1/dsl/io/__init__.py 1 tfx/v1/dsl/io/fileio.py 15 tfx/v1/dsl/placeholders/__init__.py 3 tfx/v1/extensions/__init__.py 2 tfx/v1/extensions/google_cloud_ai_platform/__init__.py 14 tfx/v1/extensions/google_cloud_ai_platform/experimental/__init__.py 5 tfx/v1/extensions/google_cloud_big_query/__init__.py 3 tfx/v1/orchestration/__init__.py 3 tfx/v1/orchestration/experimental/__init__.py 20 tfx/v1/orchestration/metadata.py 4 tfx/v1/proto/__init__.py 100 tfx/v1/types/__init__.py 2 tfx/v1/types/standard_artifacts.py 20 tfx/v1/utils/__init__.py 2 tfx/version.py 1 tfx/workspace.bzl 86