fix: remove unused environment variable (#424)

* fix: remove unused environment variable

* fix: handle nil workers in GetAllocationReqByNodeName to prevent panics

Author: 0x5457

internal/constants/env.go

@@ -68,8 +68,6 @@ const (
 	ConnectionSharedMemSize = "256"
 	ConnectionSharedMemName = "tf_shm"
 
-	RealNvmlLibPathEnv   = "TF_NVML_LIB_PATH"
-	RealCUDALibPathEnv   = "TF_CUDA_LIB_PATH"
 	RealNvmlLibPathValue = "/lib/x86_64-linux-gnu/libnvidia-ml.so.1"
 	RealCUDALibPathValue = "/lib/x86_64-linux-gnu/libcuda.so"
 

internal/gpuallocator/gpuallocator.go

@@ -642,7 +642,11 @@ func (s *GpuAllocator) DeallocByPodIdentifier(ctx context.Context, podIdentifier
 
 func (s *GpuAllocator) GetAllocationReqByNodeName(nodeName string) []*tfv1.AllocRequest {
 	allocRequests := make([]*tfv1.AllocRequest, 0, 8)
-	for workerName := range s.nodeWorkerStore[nodeName] {
+	workers, exists := s.nodeWorkerStore[nodeName]
+	if !exists || workers == nil {
+		return allocRequests
+	}
+	for workerName := range workers {
 		podUID := s.podNamespaceNsToPodUID[workerName.String()]
 		if podUID == "" {
 			continue

internal/utils/compose.go

@@ -377,12 +377,6 @@ func AddTFDefaultClientConfBeforePatch(
 			}
 
 			envList = append(envList, v1.EnvVar{
-				Name:  constants.RealNvmlLibPathEnv,
-				Value: constants.RealNvmlLibPathValue,
-			}, v1.EnvVar{
-				Name:  constants.RealCUDALibPathEnv,
-				Value: constants.RealCUDALibPathValue,
-			}, v1.EnvVar{
 				Name: constants.HypervisorIPEnv,
 				ValueFrom: &v1.EnvVarSource{
 					FieldRef: &v1.ObjectFieldSelector{