YARN-9337. Addendum to fix compilation error due to mockito spy call
(cherry picked from commit bb37c6cb7f
)
This commit is contained in:
parent
4fa0de9f04
commit
2fcbdf4131
|
@ -505,7 +505,7 @@ public class TestGpuDiscoverer {
|
||||||
Configuration conf = new Configuration();
|
Configuration conf = new Configuration();
|
||||||
conf.set(YarnConfiguration.NM_GPU_ALLOWED_DEVICES, "0:1,2:3");
|
conf.set(YarnConfiguration.NM_GPU_ALLOWED_DEVICES, "0:1,2:3");
|
||||||
|
|
||||||
GpuDiscoverer gpuSpy = spy(GpuDiscoverer.class);
|
GpuDiscoverer gpuSpy = spy(new GpuDiscoverer());
|
||||||
|
|
||||||
gpuSpy.initialize(conf);
|
gpuSpy.initialize(conf);
|
||||||
gpuSpy.getGpusUsableByYarn();
|
gpuSpy.getGpusUsableByYarn();
|
||||||
|
|
Loading…
Reference in New Issue