Skip to content

Commit de2ed5c

Browse files
committed
fix: HF estimator attach test to use latest py version instead of hardcoded version
1 parent 6f5ed2e commit de2ed5c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/unit/sagemaker/huggingface/test_estimator.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -294,7 +294,7 @@ def test_attach(
294294

295295
estimator = HuggingFace.attach(training_job_name="neo", sagemaker_session=sagemaker_session)
296296
assert estimator.latest_training_job.job_name == "neo"
297-
assert estimator.py_version == "py36"
297+
assert estimator.py_version == huggingface_pytorch_training_py_version
298298
assert estimator.framework_version == huggingface_training_version
299299
assert estimator.pytorch_version == huggingface_pytorch_training_version
300300
assert estimator.role == "arn:aws:iam::366:role/SageMakerRole"

0 commit comments

Comments
 (0)