Skip to content
This repository was archived by the owner on Apr 15, 2022. It is now read-only.

Commit f7ea54d

Browse files
committed
feature details
1 parent 792e6f3 commit f7ea54d

File tree

2 files changed

+17
-6
lines changed

2 files changed

+17
-6
lines changed

splicemachine/features/feature_store.py

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,16 @@ def get_features_by_name(self, names: Optional[List[str]] = None, as_list=False)
128128
r = make_request(self._FS_URL, Endpoints.FEATURES, RequestType.GET, self._basic_auth, { "name": names })
129129
return [Feature(**f) for f in r] if as_list else pd.DataFrame.from_dict(r)
130130

131+
def get_feature_details(self, name: str) -> Feature:
132+
"""
133+
Returns a Feature and it's detailed information
134+
135+
:param name: The feature name
136+
:return: Feature
137+
"""
138+
r = make_request(self._FS_URL, Endpoints.FEATURE_DETAILS, RequestType.GET, self._basic_auth, { "name": name })
139+
return Feature(**r)
140+
131141
def get_feature_vector(self, features: List[Union[str, Feature]],
132142
join_key_values: Dict[str, str], return_primary_keys = True, return_sql=False) -> Union[str, PandasDF]:
133143
"""
@@ -513,7 +523,7 @@ def describe_feature_sets(self) -> None:
513523
514524
:return: None
515525
"""
516-
r = make_request(self._FS_URL, Endpoints.FEATURE_SET_DESCRIPTIONS, RequestType.GET, self._basic_auth)
526+
r = make_request(self._FS_URL, Endpoints.FEATURE_SET_DETAILS, RequestType.GET, self._basic_auth)
517527

518528
print('Available feature sets')
519529
for desc in r:
@@ -535,7 +545,7 @@ def describe_feature_set(self, schema_name: str, table_name: str) -> None:
535545
schema_name = schema_name.upper()
536546
table_name = table_name.upper()
537547

538-
r = make_request(self._FS_URL, Endpoints.FEATURE_SET_DESCRIPTIONS, RequestType.GET, self._basic_auth,
548+
r = make_request(self._FS_URL, Endpoints.FEATURE_SET_DETAILS, RequestType.GET, self._basic_auth,
539549
params={'schema':schema_name, 'table':table_name})
540550
descs = r
541551
if not descs: raise SpliceMachineException(
@@ -558,7 +568,7 @@ def describe_training_views(self) -> None:
558568
:param training_view: The training view name
559569
:return: None
560570
"""
561-
r = make_request(self._FS_URL, Endpoints.TRAINING_VIEW_DESCRIPTIONS, RequestType.GET, self._basic_auth)
571+
r = make_request(self._FS_URL, Endpoints.TRAINING_VIEW_DETAILS, RequestType.GET, self._basic_auth)
562572

563573
print('Available training views')
564574
for desc in r:
@@ -575,7 +585,7 @@ def describe_training_view(self, training_view: str) -> None:
575585
:return: None
576586
"""
577587

578-
r = make_request(self._FS_URL, Endpoints.TRAINING_VIEW_DESCRIPTIONS, RequestType.GET, self._basic_auth, {'name': training_view})
588+
r = make_request(self._FS_URL, Endpoints.TRAINING_VIEW_DETAILS, RequestType.GET, self._basic_auth, {'name': training_view})
579589
descs = r
580590
if not descs: raise SpliceMachineException(f"Training view {training_view} not found. Check name and try again.")
581591
desc = descs[0]

splicemachine/features/utils/http_utils.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,9 @@ class Endpoints:
3030
"""
3131
DEPLOYMENTS: str = "deployments"
3232
FEATURES: str = "features"
33+
FEATURE_DETAILS: str = "feature-details"
3334
FEATURE_SETS: str = "feature-sets"
34-
FEATURE_SET_DESCRIPTIONS: str = "feature-set-descriptions"
35+
FEATURE_SET_DETAILS: str = "feature-set-details"
3536
DEPLOY_FEATURE_SET: str = "deploy-feature-set"
3637
FEATURE_VECTOR: str = "feature-vector"
3738
FEATURE_VECTOR_SQL: str = "feature-vector-sql"
@@ -40,7 +41,7 @@ class Endpoints:
4041
TRAINING_SET_FROM_DEPLOYMENT: str = "training-set-from-deployment"
4142
TRAINING_SET_FROM_VIEW: str = "training-set-from-view"
4243
TRAINING_VIEWS: str = "training-views"
43-
TRAINING_VIEW_DESCRIPTIONS: str = "training-view-descriptions"
44+
TRAINING_VIEW_DETAILS: str = "training-view-details"
4445
TRAINING_VIEW_FEATURES: str = "training-view-features"
4546
TRAINING_VIEW_ID: str = "training-view-id"
4647
SUMMARY: str = "summary"

0 commit comments

Comments
 (0)