while registering model I am getting error: AssertionError:
Options
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
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,
)
