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: 

ML Practioner | ML 10 - Feature Store notebook | feature_store import error

Slalom_Tobias
New Contributor III

the following code...

from pyspark.sql.functions import monotonically_increasing_id, lit, expr, rand

import uuid

from databricks import feature_store

from pyspark.sql.types import StringType, DoubleType

from databricks.feature_store import feature_table, FeatureLookup

import mlflow

import mlflow.sklearn

from mlflow.models.signature import infer_signature

from sklearn.model_selection import train_test_split

from sklearn.ensemble import RandomForestRegressor

from sklearn.metrics import mean_squared_error, r2_score

generates the following error....

ImportError: cannot import name 'feature_store' from 'databricks' (/databricks/python/lib/python3.8/site-packages/databricks/__init__.py)

please see screenshot for configuration specifics.

1 ACCEPTED SOLUTION

Accepted Solutions

Prabakar
Esteemed Contributor III
Esteemed Contributor III

@Tobias Cortese​ I could see you are using the standard DBR version and it is expected. Please use the ML DBR version and it should resolve your issue.

View solution in original post

2 REPLIES 2

Prabakar
Esteemed Contributor III
Esteemed Contributor III

@Tobias Cortese​ I could see you are using the standard DBR version and it is expected. Please use the ML DBR version and it should resolve your issue.

Anonymous
Not applicable

Hope that was an easy fix - @Tobias Cortese​ ! Thanks for marking the "best answer"!

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!