Skip to content

Commit 9626d9b

Browse files
psnileshNilesh
authored andcommitted
Create a default SageMaker Session inside FeatureGroup class (aws#3667)
Co-authored-by: Nilesh <[email protected]>
1 parent 36d0f27 commit 9626d9b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/sagemaker/feature_store/feature_group.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -479,11 +479,12 @@ class FeatureGroup:
479479
Attributes:
480480
name (str): name of the FeatureGroup instance.
481481
sagemaker_session (Session): session instance to perform boto calls.
482+
If None, a new Session will be created.
482483
feature_definitions (Sequence[FeatureDefinition]): list of FeatureDefinitions.
483484
"""
484485

485486
name: str = attr.ib(factory=str)
486-
sagemaker_session: Session = attr.ib(default=Session)
487+
sagemaker_session: Session = attr.ib(factory=Session)
487488
feature_definitions: Sequence[FeatureDefinition] = attr.ib(factory=list)
488489

489490
_INTEGER_TYPES = [

0 commit comments

Comments
 (0)