Skip to content

Commit 32c0ae0

Browse files
committed
Externalize the common logic for controllers
1 parent a6cf290 commit 32c0ae0

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pkg/controller/dual-pods/controller.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ import (
2929
"sync/atomic"
3030
"time"
3131

32+
"github.com/llm-d-incubation/llm-d-fast-model-actuation/pkg/controller/utils"
3233
corev1 "k8s.io/api/core/v1"
3334
apierrors "k8s.io/apimachinery/pkg/api/errors"
3435
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
@@ -102,7 +103,7 @@ func GPUIndexFunc(obj any) ([]string, error) {
102103
if len(pod.Annotations[nominalHashAnnotationKey]) == 0 || pod.Spec.NodeName == "" {
103104
return []string{}, nil
104105
}
105-
isIdx, _, err := getInferenceServerPort(pod)
106+
isIdx, _, err := utils.GetInferenceServerPort(pod)
106107
if err != nil {
107108
return []string{}, nil
108109
}

0 commit comments

Comments
 (0)