Skip to content

Commit 2904cd3

Browse files
authored
fix: disable profiler by default for regions not support it (#2134)
1 parent f38df4f commit 2904cd3

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/sagemaker/estimator.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -349,6 +349,9 @@ def __init__(
349349
self.profiler_config = profiler_config
350350
self.disable_profiler = disable_profiler
351351

352+
if not _region_supports_profiler(self.sagemaker_session.boto_region_name):
353+
self.disable_profiler = True
354+
352355
self.profiler_rule_configs = None
353356
self.profiler_rules = None
354357
self.debugger_rules = None

0 commit comments

Comments
 (0)