Tensorflow Extended Pusher component does not support CMEK

[crbl1122]

If the bug is related to a specific library below, please raise an issue in the respective repo directly:

TensorFlow Data Validation Repo

TensorFlow Model Analysis Repo

TensorFlow Transform Repo

TensorFlow Serving Repo

System information

• Have I specified the code to reproduce the issue (Yes, No):
• Environment in which the code is executed (e.g., Local(Linux/MacOS/Windows), Interactive Notebook, Google Cloud, etc):
• TensorFlow version:
• TFX Version:
• Python version:
• Python dependencies (from pip freeze output):

Describe the current behavior The Tensorflow Extended Pusher component does not support customer managed encryption keys (CMEK). This component does not pass the encryption_keys to its underlying jobs Upload Model (to Model Registry in Vertex AI) and model Deploy to and endpoint.

Describe the expected behavior

The TFX should accept CMEK

Standalone code to reproduce the issue

Providing a bare minimum test case or step(s) to reproduce the problem will greatly help us to debug the issue. If possible, please share a link to Colab/Jupyter/any notebook.

Name of your Organization (Optional)

Other info / logs

Include any logs or source code that would be helpful to diagnose the problem. If including tracebacks, please include the full traceback. Large logs and files should be attached.