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
Databricks Employee
Databricks Employee

@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
Databricks Employee
Databricks Employee

@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"!

Connect with Databricks Users in Your Area

Join a Regional User Group to connect with local Databricks users. Events will be happening in your city, and you won’t want to miss the chance to attend and share knowledge.

If there isn’t a group near you, start one and help create a community that brings people together.

Request a New Group