Skip registration on empty socket

Signed-off-by: Evan Lezar <elezar@nvidia.com>
NVIDIA · Jun 27, 2024 · 7e7c85d · 7e7c85d
1 parent 0a7ef7f
commit 7e7c85d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/internal/plugin/server.go b/internal/plugin/server.go
@@ -245,6 +245,11 @@ func (plugin *NvidiaDevicePlugin) Serve() error {
 
 // Register registers the device plugin for the given resourceName with Kubelet.
 func (plugin *NvidiaDevicePlugin) Register(kubeletSocket string) error {
+	if kubeletSocket == "" {
+		klog.Info("Skipping registration with Kubelet")
+		return nil
+	}
+
 	conn, err := plugin.dial(kubeletSocket, 5*time.Second)
 	if err != nil {
 		return err