aws / aws/sagemaker-python-sdk

Model does not have a sagemaker_session property

Open
#5,829 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
2.3k
Forks
1.3k
Avg merge
1d 22h
Merged PRs (30d)
35

Description

**PySDK Version**
- [ ] PySDK V2 (2.x)
- [ X] PySDK V3 (3.x)

**Describe the bug**
A clear and concise description of what the bug is.

ModelStep expects the Model class to have a `sagemaker_session` attribute but the spec does not allow it

**To reproduce**

Provide a Model instance as the `model` argument of the register model step arguments

**Expected behavior**
ModelStep can be instantiated with a repack step

**Screenshots or logs**
If applicable, add screenshots or logs to help explain your problem.

**System information**
A description of your system. Please provide:
- **SageMaker Python SDK version**: 3
- **Framework name (eg. PyTorch) or algorithm (eg. KMeans)**: SKLearn
- **Framework version**: 1.4.2
- **Python version**: 3.11
- **CPU or GPU**: CPU
- **Custom Docker image (Y/N)**: N

**Additional context**
Add any other context about the problem here.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.