diff --git a/x-pack/plugin/gpu/src/internalClusterTest/java/org/elasticsearch/plugin/gpu/TestCuVSServiceProvider.java b/x-pack/plugin/gpu/src/internalClusterTest/java/org/elasticsearch/plugin/gpu/TestCuVSServiceProvider.java index fa386bffbe3cb..efa95608c8266 100644 --- a/x-pack/plugin/gpu/src/internalClusterTest/java/org/elasticsearch/plugin/gpu/TestCuVSServiceProvider.java +++ b/x-pack/plugin/gpu/src/internalClusterTest/java/org/elasticsearch/plugin/gpu/TestCuVSServiceProvider.java @@ -25,6 +25,9 @@ public class TestCuVSServiceProvider extends CuVSServiceProvider { @Override public CuVSProvider get(CuVSProvider builtin) { + if (mockedGPUInfoProvider == null) { + return builtin; + } return new CuVSProviderDelegate(builtin) { @Override public GPUInfoProvider gpuInfoProvider() {