FeatureEngineeringClient workspace id error

Kabi
New Contributor III

Hi, I am working from local notebook using vscode databricks extension.

I am trying to use FeatureEngineeringClient, when I create data set

    training_set = fe.create_training_set(
        df=filtered_data_train,
        feature_lookups=payments_feature_lookups,
        label="churn",
        exclude_columns=exclude_columns,
    )

I receive error

	"name": "TypeError",
	"message": "'NoneType' object cannot be interpreted as an integer",

but looking at logs I see error related with

File ~/projects/databricks_hwm_churn/.venv/lib/python3.12/site-packages/databricks/ml_features_common/entities/feature_spec.py:243, in FeatureSpec.to_proto(self)
    241     proto_feature_spec.input_functions.append(function_info.to_proto())
    242 proto_feature_spec.serialization_version = self.serialization_version
--> 243 proto_feature_spec.workspace_id = self.workspace_id
    244 proto_feature_spec.feature_store_client_version = (
    245     self._feature_store_client_version
    246 )
    247 return proto_feature_spec

Could someone explain how can I fix problem with self.workspace_id working from local notebook?

Same code is working if I run it from databricks browser notebook.

Thank you!