while registering model I am getting error: AssertionError:

ml-engineer
Databricks Partner

‎04-25-2024 05:38 AM

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,
            )
Preview file
27 KB
0 Kudos

Kumaran
Databricks Employee
Databricks Employee

‎04-30-2024 11:29 AM

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?

0 Kudos