Skip to content

Commit 651b662

Browse files
committed
lint
1 parent fd9dd85 commit 651b662

File tree

1 file changed

+9
-10
lines changed

1 file changed

+9
-10
lines changed

src/sagemaker/estimator.py

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -2212,17 +2212,16 @@ def _validate_and_set_debugger_configs(self):
22122212
self.debugger_hook_config = False
22132213

22142214
# Disable debugger if checkpointing is enabled by the customer
2215-
_should_disable_debugger = False
22162215
if self.checkpoint_s3_uri and self.checkpoint_local_path and self.debugger_hook_config:
2217-
if self.instance_count > 1:
2218-
_should_disable_debugger = True
2219-
if hasattr(self, "distribution") and self.distribution is not None:
2220-
_should_disable_debugger = True
2221-
if _should_disable_debugger:
2222-
logger.info(
2223-
"SMDebug Does Not Currently Support Distributed Training Jobs With Checkpointing Enabled"
2224-
)
2225-
self.debugger_hook_config = False
2216+
if self.instance_count > 1 \
2217+
or (
2218+
hasattr(self, "distribution") and self.distribution is not None # pylint: disable=no-member
2219+
):
2220+
logger.info(
2221+
"SMDebug Does Not Currently Support \
2222+
Distributed Training Jobs With Checkpointing Enabled"
2223+
)
2224+
self.debugger_hook_config = False
22262225

22272226
def _stage_user_code_in_s3(self):
22282227
"""Upload the user training script to s3 and return the location.

0 commit comments

Comments
 (0)