topic Re: ML Practioner | ML 10 - Feature Store notebook | feature_store import error in Machine Learning

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

Slalom_Tobias — Mon, 01 Aug 2022 17:23:18 GMT

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.

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

Prabakar — Mon, 01 Aug 2022 22:31:21 GMT

@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.

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

Anonymous — Wed, 03 Aug 2022 18:06:42 GMT

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