Skip to content

Commit 95b88db

Browse files
author
Drew Keller
committed
fix: Remove tags from TransformJob creation in Transformer
The TransformJob resource doesn't accept tags parameter during initialization. Remove tags from the transformed dict before passing to TransformJob constructor to prevent errors.
1 parent 1572b32 commit 95b88db

2 files changed

Lines changed: 21 additions & 0 deletions

File tree

‎sagemaker-core/src/sagemaker/core/transformer.py‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -409,6 +409,8 @@ def submit(request):
409409
from sagemaker.core.utils.code_injection.codec import transform as transform_util
410410

411411
transformed = transform_util(serialized_request, "CreateTransformJobRequest")
412+
# Remove tags from transformed dict as TransformJob resource doesn't accept it
413+
transformed.pop("tags", None)
412414
self.latest_transform_job = TransformJob(**transformed)
413415

414416
if wait:

‎sagemaker-core/tests/unit/test_transformer.py‎

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -639,3 +639,22 @@ def test_load_config(self, mock_session):
639639
assert "resource_config" in config
640640
assert config["output_config"]["s3_output_path"] == "s3://bucket/output"
641641
assert config["resource_config"]["instance_count"] == 2
642+
643+
def test_transform_removes_tags_from_transform_job(self, mock_session):
644+
"""Test that tags are removed from transformed dict before TransformJob creation"""
645+
transformer = Transformer(
646+
model_name="test-model",
647+
instance_count=1,
648+
instance_type="ml.m5.xlarge",
649+
sagemaker_session=mock_session,
650+
tags=[{"Key": "test", "Value": "value"}],
651+
)
652+
653+
with patch("sagemaker.core.transformer.TransformJob") as mock_job_class:
654+
transformer.transform(
655+
data="s3://bucket/input",
656+
job_name="test-job",
657+
)
658+
mock_job_class.assert_called_once()
659+
call_kwargs = mock_job_class.call_args[1]
660+
assert "tags" not in call_kwargs

0 commit comments

Comments
 (0)