Join the SIG TFX-Addons community and help make TFX even better!


A TFX component to push validated TensorFlow models to a model serving platform.

Used in the notebooks

Used in the tutorials

The Pusher component can be used to push an validated SavedModel from output of the Trainer component to TensorFlow Serving. The Pusher will check the validation results from the Evaluator component and InfraValidator component before deploying the model. If the model has not been blessed, then the model will not be pushed.


  # Checks whether the model passed the validation steps and pushes the model
  # to a file destination if check passed.
  pusher = Pusher(

Component outputs contains:

See the Pusher guide for more details.

model An optional Channel of type standard_artifacts.Model, usually produced by a Trainer component.
model_blessing An optional Channel of type standard_artifacts.ModelBlessing, usually produced from an Evaluator component.
infra_blessing An optional Channel of type standard_artifacts.InfraBlessing, usually produced from an InfraValidator component.
push_destination A pusher_pb2.PushDestination instance, providing info for tensorflow serving to load models. Optional if executor_class doesn't require push_destination. If any field is provided as a RuntimeParameter, push_destination should be constructed as a dict with the same field names as PushDestination proto message.
custom_config A dict which contains the deployment job parameters to be passed to Cloud platforms.
custom_executor_spec Optional custom executor spec. This is experimental and is subject to change in the future.

outputs Component's output channel dict.