Skip to content

Commit 3d82d57

Browse files
committed
Update integration tests for pipeline training step
To verify the profiler is enabled by default for the training step
1 parent a8f48da commit 3d82d57

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/integ/test_workflow.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -450,7 +450,7 @@ def test_conditional_pytorch_training_model_registration(
450450
pass
451451

452452

453-
def test_training_job_with_debugger(
453+
def test_training_job_with_debugger_and_profiler(
454454
sagemaker_session,
455455
pipeline_name,
456456
role,
@@ -535,6 +535,9 @@ def test_training_job_with_debugger(
535535
config["RuleParameters"]["rule_to_invoke"] == rule.rule_parameters["rule_to_invoke"]
536536
)
537537
assert job_description["DebugHookConfig"] == debugger_hook_config._to_request_dict()
538+
539+
assert job_description["ProfilingStatus"] == "Enabled"
540+
assert job_description["ProfilerConfig"]["ProfilingIntervalInMilliseconds"] == 500
538541
finally:
539542
try:
540543
pipeline.delete()

0 commit comments

Comments
 (0)