File tree Expand file tree Collapse file tree 2 files changed +9
-3
lines changed Expand file tree Collapse file tree 2 files changed +9
-3
lines changed Original file line number Diff line number Diff line change @@ -72,4 +72,4 @@ def partition(self):
72
72
73
73
@partition .setter
74
74
def partition (self , partition ):
75
- self ._partition
75
+ self ._partition = partition
Original file line number Diff line number Diff line change @@ -84,8 +84,14 @@ def __init__(
84
84
self ._default_storage = default_storage
85
85
self ._hudi_enabled = hudi_enabled
86
86
87
- self ._primary_key = primary_key
88
- self ._partition_key = partition_key
87
+ if id is None :
88
+ # Initialized from the API
89
+ self ._primary_key = primary_key
90
+ self ._partition_key = partition_key
91
+ else :
92
+ # Initialized from the backend
93
+ self ._primary_key = [f .name for f in self ._features if f .primary ]
94
+ self ._partition_key = [f .name for f in self ._features if f .partition ]
89
95
90
96
self ._feature_group_engine = feature_group_engine .FeatureGroupEngine (
91
97
featurestore_id
You can’t perform that action at this time.
0 commit comments