Fixing logic error in training_compiler

Lokiiiiii · Lokiiiiii · commit 56ca8802b868 · 2022-06-15T10:06:47.000-07:00
diff --git a/tests/unit/sagemaker/training_compiler/test_tensorflow_compiler.py b/tests/unit/sagemaker/training_compiler/test_tensorflow_compiler.py
@@ -93,7 +93,7 @@ def _get_full_gpu_image_uri(framework_version, instance_type, training_compiler_
         py_version=py_version,
         instance_type=instance_type,
         image_scope="training",
-        container_version="cu112-ubuntu20.04",
+        container_version=None,
         training_compiler_config=training_compiler_config,
     )
 

Original file line number	Diff line number	Diff line change
`@@ -93,7 +93,7 @@ def _get_full_gpu_image_uri(framework_version, instance_type, training_compiler_`
`93`	`93`	`py_version=py_version,`
`94`	`94`	`instance_type=instance_type,`
`95`	`95`	`image_scope="training",`
`96`		`- container_version="cu112-ubuntu20.04",`
	`96`	`+ container_version=None,`
`97`	`97`	`training_compiler_config=training_compiler_config,`
`98`	`98`	`)`
`99`	`99`