fix: remove nvidia.com/gpu resource requests and limits from containers (#140)

Author: 0x5457

internal/constants/constants.go

@@ -2,6 +2,9 @@ package constants
 
 import "time"
 
+const (
+	NvidiaGPUKey = "nvidia.com/gpu"
+)
 const (
 	// Domain is the domain prefix used for all tensor-fusion.ai related annotations and finalizers
 	Domain = "tensor-fusion.ai"

internal/webhook/v1/pod_webhook.go

@@ -255,6 +255,14 @@ func (m *TensorFusionPodMutator) patchTFClient(
 					return nil, fmt.Errorf("unmarshal patched container: %w", err)
 				}
 
+				// remove nvidia.com/gpu in resources
+				if container.Resources.Requests != nil {
+					delete(container.Resources.Requests, constants.NvidiaGPUKey)
+				}
+				if container.Resources.Limits != nil {
+					delete(container.Resources.Limits, constants.NvidiaGPUKey)
+				}
+
 				// add connection env
 				connectionName := fmt.Sprintf("%s%s", pod.GenerateName, shortuuid.NewWithAlphabet("123456789abcdefghijkmnopqrstuvwxy"))
 				connectionNamespace := pod.Namespace