Skip to content

Commit c35de7b

Browse files
psnileshNilesh
and
Nilesh
authored
Create a default SageMaker Session inside FeatureGroup class (#3667)
Co-authored-by: Nilesh <[email protected]>
1 parent cc6a358 commit c35de7b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/sagemaker/feature_store/feature_group.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -473,11 +473,12 @@ class FeatureGroup:
473473
Attributes:
474474
name (str): name of the FeatureGroup instance.
475475
sagemaker_session (Session): session instance to perform boto calls.
476+
If None, a new Session will be created.
476477
feature_definitions (Sequence[FeatureDefinition]): list of FeatureDefinitions.
477478
"""
478479

479480
name: str = attr.ib(factory=str)
480-
sagemaker_session: Session = attr.ib(default=Session)
481+
sagemaker_session: Session = attr.ib(factory=Session)
481482
feature_definitions: Sequence[FeatureDefinition] = attr.ib(factory=list)
482483

483484
_INTEGER_TYPES = [

0 commit comments

Comments
 (0)