aws / aws/sagemaker-python-sdk

Make torch an optional dependency / extra

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

Description

Modern packages use optional dependecies / extras for specific functionality. I assume this is very well possible for the sagemaker sdk.

A similar request was posted in discussions: https://github.com/aws/sagemaker-python-sdk/discussions/5441

I couldn't find any actual "issue" and hope this gets more visibility.

The main and most urgent topic is the dependency of pytorch which requires 354/620mb size in a clean venv. I would assume that most use cases don't need pytorch at all.

Contributor guide

Open the contributing guide

Research direction

No files or tests are named in the issue. Start by locating the SDK's dependency declarations and installation/extras configuration, then trace where PyTorch is imported or required; compare a clean install with the proposed extra. Done means the base package installs without PyTorch while PyTorch-dependent SageMaker functionality remains usable through an explicit extra.

Written by the indexing model from the issue text.

Assessment

Tech stack
aws, python, pytorch
Domain
build-system, machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Quiet
Clarity
Mostly clear
Newbie friendliness
50/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.