Skip to content

Commit d83851a

Browse files
committed
fixing unit test for PT 2.3 change
1 parent 6e0e833 commit d83851a

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/unit/sagemaker/image_uris/test_smp_v2.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,9 @@ def test_smp_v2(load_config):
3838
if "2.1" in version or "2.2" in version or "2.3" in version:
3939
cuda_vers = "cu121"
4040

41+
if "2.3" in version:
42+
py_version = "py311"
43+
4144
uri = image_uris.get_training_image_uri(
4245
region,
4346
framework="pytorch",

0 commit comments

Comments
 (0)