We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6f5ed2e commit de2ed5cCopy full SHA for de2ed5c
tests/unit/sagemaker/huggingface/test_estimator.py
@@ -294,7 +294,7 @@ def test_attach(
294
295
estimator = HuggingFace.attach(training_job_name="neo", sagemaker_session=sagemaker_session)
296
assert estimator.latest_training_job.job_name == "neo"
297
- assert estimator.py_version == "py36"
+ assert estimator.py_version == huggingface_pytorch_training_py_version
298
assert estimator.framework_version == huggingface_training_version
299
assert estimator.pytorch_version == huggingface_pytorch_training_version
300
assert estimator.role == "arn:aws:iam::366:role/SageMakerRole"
0 commit comments