while registering model I am getting error: AssertionError:

ml-engineer — Thu, 25 Apr 2024 12:38:32 GMT

I am getting error while running the code with workflow if I running code individually with notebook then its running fine.

below is the code :

fe = FeatureEngineeringClient()

self.logger.info("Logging model to MLflow using Feature Engineering API")

fe.log_model(

model=SklearnModelWrapper(model),

artifact_path="fe_model",

flavor=mlflow.pyfunc,

training_set=fe_training_set,

registered_model_name=registered_model_name,

)

Re: while registering model I am getting error: AssertionError:

Kumaran — Tue, 30 Apr 2024 18:29:52 GMT

Hi @ml-engineer Thank you for contacting Databricks community support.

I think you are installing tensorflow version 2.14, could you update to 2.16 using the pip install --upgrade tensorflow and see how it works?