cancel
Showing results for 
Search instead for 
Did you mean: 
Machine Learning
Dive into the world of machine learning on the Databricks platform. Explore discussions on algorithms, model training, deployment, and more. Connect with ML enthusiasts and experts.
cancel
Showing results for 
Search instead for 
Did you mean: 

while registering model I am getting error: AssertionError:

ml-engineer
New Contributor

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,
            )
1 REPLY 1

Kumaran
Valued Contributor III
Valued Contributor III

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?

Join 100K+ Data Experts: Register Now & Grow with Us!

Excited to expand your horizons with us? Click here to Register and begin your journey to success!

Already a member? Login and join your local regional user group! If there isn’t one near you, fill out this form and we’ll create one for you to join!