From c3ba6ae03361f0c3c5e2da102e9fa909c2046975 Mon Sep 17 00:00:00 2001 From: ishaansehgal99 Date: Thu, 4 Apr 2024 14:38:08 -0700 Subject: [PATCH] remove cycle --- pkg/utils/common-preset.go | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/pkg/utils/common-preset.go b/pkg/utils/common-preset.go index a1a514413..ac0005577 100644 --- a/pkg/utils/common-preset.go +++ b/pkg/utils/common-preset.go @@ -5,9 +5,7 @@ package utils import ( "fmt" kaitov1alpha1 "github.com/azure/kaito/api/v1alpha1" - "github.com/azure/kaito/pkg/tuning" corev1 "k8s.io/api/core/v1" - "strconv" ) const ( @@ -70,11 +68,7 @@ func GetInstanceGPUCount(wObj *kaitov1alpha1.Workspace) int { sku := wObj.Resource.InstanceType gpuConfig, exists := kaitov1alpha1.SupportedGPUConfigs[sku] if !exists { - numProcesses, err := strconv.Atoi(tuning.DefaultNumProcesses) - if err != nil { - return 1 - } - return numProcesses + return 1 } return gpuConfig.GPUCount }