aws / aws/sagemaker-python-sdk

Local processing job replaces local output location with S3 URI

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

描述

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

**Describe the bug**
When using `LocalSession` with `FrameworkProcessor`, the SDK ignores `file://` URIs specified in `ProcessingOutput.s3_output.s3_uri` and replaces them with S3 URIs. This prevents local processing jobs from saving outputs to local directories as intended.

The issue is in `Processor._normalize_outputs()` (line 489 in `sagemaker-core/src/sagemaker/core/processing.py`), which replaces any non-S3 URI with an S3 URI without checking if the session is a `LocalSession` that should preserve `file://` URIs.

**To reproduce**
```python
from sagemaker.core import FrameworkProcessor
from sagemaker.core.local import LocalSession
from sagemaker.core.shapes import ProcessingOutput, ProcessingS3Output
from sagemaker.core.image_uris import retrieve
import os

# Create local session
local_session = LocalSession()

# Get processor image
processor_image_uri = retrieve(
framework="sklearn",
version="1.4-2",
region=local_session.boto_region_name
)

# Define outputs with file:// URIs
local_processing_dir = os.path.abspath("processing")
os.makedirs(f"{local_processing_dir}/train", exist_ok=True)

local_processing_outputs = [
ProcessingOutput(
output_name="train",
s3_output=ProcessingS3Output(
s3_uri=f"file://{local_processing_dir}/train",
local_path="/opt/ml/processing/output/train",
s3_upload_mode="EndOfJob")
)
]

# Create processor with LocalSession
processor = FrameworkProcessor(
image_uri=processor_image_uri,
role="arn:aws:iam::123456789012:role/DummyRole",
instance_type="local",
instance_count=1,
sagemaker_session=local_session,
base_job_name='test-local-processing',
command=["python3"]
)

# Create a simple processing script
os.makedirs("processing_code", exist_ok=True)
with open("processing_code/test.py", "w") as f:
f.write("""
import os
with open('/opt/ml/processing/output/train/output.txt', 'w') as f:
f.write('test output')
print('Processing complete')
""")

# Run processor
processor.run(
code="test.py",
source_dir="./processing_code",
outputs=local_processing_outputs,
wait=False,
logs=True
)

# Check where outputs went
print(f"Expected output location: {local_processing_dir}/train/output.txt")
print(f"File exists locally: {os.path.exists(f'{local_processing_dir}/train/output.txt')}")
```

**Expected behavior**
When using LocalSession with `file://` URIs in `ProcessingOutput`, the outputs should be saved to the specified local directories, not uploaded to S3.

**Screenshots or logs**
N/A

**System information**
- **SageMaker Python SDK version**: 3.4.0 (sagemaker-core 2.4.0)
- **Framework name (eg. PyTorch) or algorithm (eg. KMeans)**: scikit-learn
- **Framework version**: 1.4-2
- **Python version**: 3.10
- **CPU or GPU**: CPU
- **Custom Docker image (Y/N)**: N

**Additional context**
N/A

贡献指南

打开贡献指南

调研方向

阅读 sagemaker-core/src/sagemaker/core/processing.py 中的 Processor._normalize_outputs(),然后使用 LocalSession、FrameworkProcessor 和一个 file:// ProcessingOutput URI 重现该问题。完成的标准是本地处理保留指定的 file:// 输出位置,而不是将其替换为 S3 URI。

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

评估

技术栈
aws, python
领域
cloud, machine-learning
Issue 类型
缺陷
难度
2/5
预计耗时
1-3 小时
活跃度
停滞
描述清晰度
描述清楚
新手友好度
45/100

把新 issue 发到你的邮箱

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