kemko/nomad
1
0
Fork 0
mirror of https://github.com/kemko/nomad.git synced 2026-01-05 09:55:44 +03:00
Code Issues Packages Projects Releases Wiki Activity

removes nvidia import from docker test (#8312)

Browse Source
This commit is contained in:
Drew Bailey
2020-06-30 09:34:59 -04:00
committed by GitHub
parent dcb1324147
commit 42c2ee4448
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
drivers/docker/driver_test.go
View File

@@ -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")