Skip to content

Commit 9b98e50

Browse files
author
Verdi March
committed
Fix FrameworkProcessor to call underlying estimator with instance_count=1
This fix specific bugs with SKLearnEstimator which raises exception on instance_count > 1.
1 parent dd7a869 commit 9b98e50

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/processing.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1355,7 +1355,7 @@ def _pre_init_normalization(
13551355
entry_point="",
13561356
role="",
13571357
enable_network_isolation=False,
1358-
instance_count=instance_count,
1358+
instance_count=1,
13591359
sagemaker_session=sagemaker_session,
13601360
)
13611361
image_uri = est.training_image_uri()
@@ -1518,7 +1518,7 @@ def _upload_payload(
15181518
image_uri=self.image_uri, # The image uri is already normalized by this point.
15191519
role=self.role,
15201520
instance_type=self.instance_type,
1521-
instance_count=self.instance_count,
1521+
instance_count=1,
15221522
sagemaker_session=self.sagemaker_session,
15231523
debugger_hook_config=False,
15241524
disable_profiler=True,

0 commit comments

Comments
 (0)