topic Re: ModuleNotFoundError: No module named 'databricks.feature_store.mlflow_model' in Machine Learning

ModuleNotFoundError: No module named 'databricks.feature_store.mlflow_model'

SOlivero — Tue, 08 Aug 2023 00:19:09 GMT

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))

Re: ModuleNotFoundError: No module named 'databricks.feature_store.mlflow_model'

SOlivero — Tue, 08 Aug 2023 01:10:20 GMT

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

Re: ModuleNotFoundError: No module named 'databricks.feature_store.mlflow_model'

Kumaran — Mon, 14 Aug 2023 06:53:35 GMT

Hi @SOlivero,

Thank you for posting the question in Databricks community.

Please share with us the error message for further analysis.

Re: ModuleNotFoundError: No module named 'databricks.feature_store.mlflow_model'

hollybaker354 — Thu, 17 Aug 2023 10:24:21 GMT

Hi @SOlivero I'm having the same issue, have you found a solution?

Re: ModuleNotFoundError: No module named 'databricks.feature_store.mlflow_model'

hollybaker354 — Thu, 17 Aug 2023 10:25:28 GMT

Re: ModuleNotFoundError: No module named 'databricks.feature_store.mlflow_model'

Kumaran — Thu, 17 Aug 2023 14:20:45 GMT

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?

Re: ModuleNotFoundError: No module named 'databricks.feature_store.mlflow_model'

Kumaran — Thu, 17 Aug 2023 14:23:22 GMT

Hi @SOlivero and @hollybaker354,

Can you check by setting the maxDepth to the default value and see how it works?

Re: ModuleNotFoundError: No module named 'databricks.feature_store.mlflow_model'

x_y — Tue, 07 Nov 2023 16:28:24 GMT

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 messagelog_model artifacts