File tree Expand file tree Collapse file tree 2 files changed +9
-8
lines changed Expand file tree Collapse file tree 2 files changed +9
-8
lines changed Original file line number Diff line number Diff line change @@ -838,14 +838,14 @@ def _validate_and_set_debugger_configs(self):
838
838
if region_supports_debugger :
839
839
if self .debugger_hook_config in [None , {}]:
840
840
self .debugger_hook_config = DebuggerHookConfig (s3_output_path = self .output_path )
841
- else :
842
- if self .debugger_hook_config is not False and self .debugger_hook_config :
843
- # when user set debugger config in a unsupported region
844
- raise ValueError (
845
- "Current region does not support debugger but debugger hook config is set!"
846
- )
847
- # disable debugger in unsupported regions
848
- self .debugger_hook_config = False
841
+ else :
842
+ if self .debugger_hook_config is not False and self .debugger_hook_config :
843
+ # when user set debugger config in a unsupported region
844
+ raise ValueError (
845
+ "Current region does not support debugger but debugger hook config is set!"
846
+ )
847
+ # disable debugger in unsupported regions
848
+ self .debugger_hook_config = False
849
849
850
850
def _prepare_debugger_rules (self ):
851
851
"""Set any necessary values in debugger rules, if they are provided."""
Original file line number Diff line number Diff line change @@ -560,6 +560,7 @@ def test_region_supports_debugger_feature_returns_false_for_unsupported_regions(
560
560
assert fw_utils ._region_supports_debugger ("eu-central-2" ) is False
561
561
assert fw_utils ._region_supports_debugger ("us-gov-east-1" ) is False
562
562
563
+
563
564
def test_warn_if_parameter_server_with_multi_gpu (caplog ):
564
565
instance_type = "ml.p2.8xlarge"
565
566
distribution = {"parameter_server" : {"enabled" : True }}
You can’t perform that action at this time.
0 commit comments