ModuleNotFoundError: No module named 'databricks.feature_store.mlflow_model'
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
08-07-2023 05:19 PM
I'm using runtime DBR 12.2 LTS ML
Getting this error when running.
import mlflow
logged_model = 'runs:/.../model' #my run id
# Load model as a PyFuncModel.
loaded_model = mlflow.pyfunc.load_model(logged_model)
# Predict on a Pandas DataFrame.
import pandas as pd
loaded_model.predict(pd.DataFrame(data))
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
08-07-2023 06:10 PM
In mi logged model the MLModel has the following flavor:
flavors:
python_function:
data: data/feature_store
env:
conda: conda.yaml
virtualenv: python_env.yaml
loader_module: databricks.feature_store.mlflow_model
python_version: 3.9.5
I think here is where the problem is, but I don't know how to solve it
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
08-13-2023 11:53 PM
Hi @SOlivero,
Thank you for posting the question in Databricks community.
Please share with us the error message for further analysis.
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
08-17-2023 03:24 AM
Hi @SOlivero I'm having the same issue, have you found a solution?
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
08-17-2023 03:25 AM
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
08-17-2023 07:20 AM
Hi @hollybaker354 and @SOlivero,
Thank you for the reply.
Can you check by setting the maxDepth to the default value and let us know how it works?
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
08-17-2023 07:23 AM
Hi @SOlivero and @hollybaker354,
Can you check by setting the maxDepth to the default value and see how it works?
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
11-07-2023 08:27 AM - edited 11-07-2023 08:28 AM
hi @Kumaran
Could you please let me know where to set the "maxDepth" to the default value? What do you mean?
I'm having the same issue. Below please see the error, the artifacts of the logged model, and how I logged the model with feature store.
fs.log_model()
error message
log_model artifacts

