while registering model I am getting error: AssertionError:

[ml-engineer]

while registering model I am getting error: AssertionError:

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,

            )

[Kumaran]

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?