Skip to content

Commit 42592e7

Browse files
committed
fix: reset default output path on create transform job
1 parent d50b24f commit 42592e7

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/sagemaker/transformer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,7 @@ def __init__(
9090
self.base_transform_job_name = base_transform_job_name
9191
self._current_job_name = None
9292
self.latest_transform_job = None
93+
self._reset_output_path = False
9394

9495
self.sagemaker_session = sagemaker_session or Session()
9596

@@ -146,10 +147,11 @@ def transform(
146147

147148
self._current_job_name = name_from_base(base_name)
148149

149-
if self.output_path is None:
150+
if self.output_path is None or self._reset_output_path is True:
150151
self.output_path = "s3://{}/{}".format(
151152
self.sagemaker_session.default_bucket(), self._current_job_name
152153
)
154+
self._reset_output_path = True
153155

154156
self.latest_transform_job = _TransformJob.start_new(
155157
self,

0 commit comments

Comments
 (0)