We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a8f48da commit 3d82d57Copy full SHA for 3d82d57
tests/integ/test_workflow.py
@@ -450,7 +450,7 @@ def test_conditional_pytorch_training_model_registration(
450
pass
451
452
453
-def test_training_job_with_debugger(
+def test_training_job_with_debugger_and_profiler(
454
sagemaker_session,
455
pipeline_name,
456
role,
@@ -535,6 +535,9 @@ def test_training_job_with_debugger(
535
config["RuleParameters"]["rule_to_invoke"] == rule.rule_parameters["rule_to_invoke"]
536
)
537
assert job_description["DebugHookConfig"] == debugger_hook_config._to_request_dict()
538
+
539
+ assert job_description["ProfilingStatus"] == "Enabled"
540
+ assert job_description["ProfilerConfig"]["ProfilingIntervalInMilliseconds"] == 500
541
finally:
542
try:
543
pipeline.delete()
0 commit comments