From 42c2ee4448c8e15fd977eb38a3c4ef6d74b4e786 Mon Sep 17 00:00:00 2001 From: Drew Bailey <2614075+drewbailey@users.noreply.github.com> Date: Tue, 30 Jun 2020 09:34:59 -0400 Subject: [PATCH] removes nvidia import from docker test (#8312) --- drivers/docker/driver_test.go | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/drivers/docker/driver_test.go b/drivers/docker/driver_test.go index 17cc9dc20..9ba0e8e4a 100644 --- a/drivers/docker/driver_test.go +++ b/drivers/docker/driver_test.go @@ -17,7 +17,6 @@ import ( hclog "github.com/hashicorp/go-hclog" "github.com/hashicorp/nomad/client/taskenv" "github.com/hashicorp/nomad/client/testutil" - "github.com/hashicorp/nomad/devices/gpu/nvidia" "github.com/hashicorp/nomad/helper/freeport" "github.com/hashicorp/nomad/helper/pluginutils/hclspecutils" "github.com/hashicorp/nomad/helper/pluginutils/hclutils" @@ -959,7 +958,7 @@ func TestDockerDriver_CreateContainerConfig_RuntimeConflict(t *testing.T) { task, cfg, ports := dockerTask(t) defer freeport.Return(ports) - task.DeviceEnv[nvidia.NvidiaVisibleDevices] = "GPU_UUID_1" + task.DeviceEnv["NVIDIA_VISIBLE_DEVICES"] = "GPU_UUID_1" require.NoError(t, task.EncodeConcreteDriverConfig(cfg)) @@ -1202,7 +1201,7 @@ func TestDockerDriver_CreateContainerConfigWithRuntimes(t *testing.T) { driver.gpuRuntime = testCase.gpuRuntimeSet driver.config.GPURuntimeName = testCase.expectedRuntime if testCase.nvidiaDevicesProvided { - task.DeviceEnv[nvidia.NvidiaVisibleDevices] = "GPU_UUID_1" + task.DeviceEnv["NVIDIA_VISIBLE_DEVICES"] = "GPU_UUID_1" } c, err := driver.createContainerConfig(task, cfg, "org/repo:0.1")