[rsareddy0329]

Issue

The codestyle-doc-tests PR check is failing on master for sagemaker-core, sagemaker-mlops, and sagemaker-train (sagemaker-serve passes). The failures were introduced by earlier commits that landed .py files without running black, plus one unused import.

Reproduced on latest master with the CI-pinned tools (black==26.3.1, flake8==7.1.2):

• sagemaker-core — black-check: 5 files (workflow/job_config_document.py, workflow/pipeline_capture.py, and 3 tests/unit/workflow/ tests)
• sagemaker-mlops — black-check: 3 files (workflow/steps.py, tests/integ/workflow/test_job_step_mtrl.py, tests/unit/workflow/test_job_step.py)
• sagemaker-train — black-check: tests/unit/train/test_multi_turn_rl_trainer.py, and flake8 F401 (unused unittest.mock.PropertyMock) in the same file

The CI build log confirms flake8, pylint, docstyle, twine otherwise succeeded; only black-check (and the train F401) failed.

Description of changes

• Ran black (line-length 100) on the unformatted files in all three packages.
• Removed the unused PropertyMock import in test_multi_turn_rl_trainer.py.

Formatting-only (black preserves the AST); the sole semantic change is the unused-import removal.

Testing done

With the CI-pinned tools, in each of the three packages:

• black --check ./ → 0 files would be reformatted
• flake8 → rc 0
• pydocstyle src/sagemaker → rc 0

Merge Checklist

• I have read the CONTRIBUTING doc
• I used the commit message format described in CONTRIBUTING
• I have passed the region in to all S3 and STS clients that I've initialized as part of this change (N/A — no client changes)
• I have updated any necessary documentation (N/A — formatting only)

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.

[github-actions]

🤖 AI code review

This PR applies black formatting to several previously-unformatted .py files across sagemaker-core, sagemaker-mlops, and sagemaker-train, and removes an unused PropertyMock import to fix a flake8 F401 error. The changes are formatting-only (AST-preserving) plus the single unused-import removal, which I confirmed is unused elsewhere in the file. No issues found.

_{Reviewed commit 2a3cf98. Automated review; verify before acting.}