cancel
Showing results for 
Search instead for 
Did you mean: 
Machine Learning
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"!

Welcome to Databricks Community: Lets learn, network and celebrate together

Join our fast-growing data practitioner and expert community of 80K+ members, ready to discover, help and collaborate together while making meaningful connections. 

Click here to register and join today! 

Engage in exciting technical discussions, join a group with your peers and meet our Featured Members.