aws / aws/sagemaker-python-sdk

Make torch an optional dependency / extra

未关闭
#5,531 5 条评论 7 个 reaction 已指派 0 人 在 GitHub 查看
主要语言
Python
星标
2.3k
派生
1.3k
平均合并
1 天 22 小时
30 天内合并 PR
35

描述

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.

贡献指南

打开贡献指南

调研方向

Issue 中没有指定文件或测试。首先定位 SDK 的依赖声明和安装/extras 配置,然后追踪 PyTorch 在哪里被导入或要求;将一次干净安装与建议的 extra 进行比较。完成的标准是:基础包可以在不安装 PyTorch 的情况下安装,同时依赖 PyTorch 的 SageMaker 功能仍可通过显式 extra 使用。

由索引模型根据 Issue 内容生成。

评估

技术栈
aws, python, pytorch
领域
build-system, machine-learning
Issue 类型
功能
难度
4/5
预计耗时
3-5 天
活跃度
冷清
描述清晰度
基本清楚
新手友好度
50/100

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。