def select_except(self, features: List[Union[str, feature.Feature]] = []):
"""Select all features of the feature group except a few and return a query
object.
The query can be used to construct joins of feature groups or create a training
dataset with a subset of features of the feature group.
# Arguments
features: list, optional. A list of `Feature` objects or feature names as
strings to be selected, defaults to [], selecting all features.
# Returns
`Query`: A query object with the selected features of the feature group.
"""
if features:
except_features = [
f.name if isinstance(f, feature.Feature) else f
for f in features
]
return query.Query(
left_feature_group=self,
left_features=[
f for f in self._features if f.name not in except_features
],
feature_store_name=self._feature_store_name,
feature_store_id=self._feature_store_id,
)
else:
return self.select_all()
def select_all(self):
"""Select all features in the feature group and return a query object.
The query can be used to construct joins of feature groups or create a
training dataset immediately.
# Returns
`Query`. A query object with all features of the feature group.
"""
return query.Query(
left_feature_group=self,
left_features=self._features,
feature_store_name=self._feature_store_name,
feature_store_id=self._feature_store_id,
)
def select(self, features: List[Union[str, feature.Feature]] = []):
"""Select a subset of features of the feature group and return a query object.
The query can be used to construct joins of feature groups or create a training
dataset with a subset of features of the feature group.
# Arguments
features: list, optional. A list of `Feature` objects or feature names as
strings to be selected, defaults to [].
# Returns
`Query`: A query object with the selected features of the feature group.
"""
return query.Query(
left_feature_group=self,
left_features=features,
feature_store_name=self._feature_store_name,
feature_store_id=self._feature_store_id,
)