From 0132fc435cd7f321ed6635132da895df884f3d04 Mon Sep 17 00:00:00 2001 From: Mayank Kumar Date: Thu, 23 Mar 2023 11:19:02 -0700 Subject: [PATCH] set up base path for feature set in load function --- .../azure/ai/ml/entities/_assets/_artifacts/feature_set.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_assets/_artifacts/feature_set.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_assets/_artifacts/feature_set.py index 559fb21f102f..16430e9f89d1 100644 --- a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_assets/_artifacts/feature_set.py +++ b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_assets/_artifacts/feature_set.py @@ -139,12 +139,14 @@ def _load( ) -> "_FeatureSet": data = data or {} params_override = params_override or [] + base_path = Path(yaml_path).parent if yaml_path else Path("./") context = { - BASE_PATH_CONTEXT_KEY: Path(yaml_path).parent if yaml_path else Path("./"), + BASE_PATH_CONTEXT_KEY: base_path, PARAMS_OVERRIDE_KEY: params_override, } loaded_schema = load_from_dict(FeatureSetSchema, data, context, **kwargs) - return _FeatureSet(**loaded_schema) + feature_set = _FeatureSet(base_path=base_path, **loaded_schema) + return feature_set def _to_dict(self) -> Dict: # pylint: disable=no-member