Hi @Mathieu, earlier you wrote:



 Mathieu:

A similar error can be met when loading a TF-DF SavedModel in side binaries (e.g. the TFX evaluator [depending on the version of TFX] or the C++ TF Serving infrastructure). In this case, you will have to do the op injection:

In C++ binaries, the OP can be injected by adding a dependency to //tensorflow_decision_forests/tensorflow/ops/inference:kernel_and_op .
In Python binaries, the op can be injected by importing TF-DF (i.e. import tensorflow_decision_forests) before loading the model.



I’m trying to get a TFX pipeline to run in Vertex AI as a custom training job (TFX 1.5.0, TFDF 0.2.1, TF 2.7.0) but it’s failing in the Evaluator with this error:
tensorflow.python.framework.errors_impl.NotFoundError: Op type not registered 'SimpleMLLoadModelFromPathWithHandle' in binary running on ...
It runs locally just fine but not in Vertex AI. Could you elaborate on how I could import TF-DF in the Evaluator component?
Thank you,
Ed