Skip to content

Commit cc5abca

Browse files
committed
revert test_install_gpu_cuda_nvidia_with_spark_job cuda versions
1 parent 57fef50 commit cc5abca

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

gpu/test_gpu.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -370,10 +370,10 @@ def test_gpu_allocation(self, configuration, master_accelerator,
370370
self.verify_instance_spark()
371371

372372
@parameterized.parameters(
373-
# ("SINGLE", ["m"], GPU_T4, None, "11.8"),
374-
("STANDARD", ["m"], GPU_T4, None, "12.0"),
375-
# ("STANDARD", ["m", "w-0", "w-1"], GPU_T4, GPU_T4, "12.4"),
376-
("STANDARD", ["w-0", "w-1"], None, GPU_T4, "11.8"),
373+
("SINGLE", ["m"], GPU_T4, None, "11.8"),
374+
# ("STANDARD", ["m"], GPU_T4, None, "12.0"),
375+
("STANDARD", ["m", "w-0", "w-1"], GPU_T4, GPU_T4, "12.4"),
376+
# ("STANDARD", ["w-0", "w-1"], None, GPU_T4, "11.8"),
377377
# ("STANDARD", ["w-0", "w-1"], None, GPU_T4, "12.0"),
378378
)
379379
def test_install_gpu_cuda_nvidia_with_spark_job(self, configuration, machine_suffixes,

0 commit comments

Comments
 (0)