Fix scheduler plugin arguments readin bug.

Signed-off-by: jiangkaihua <jiangkaihua1@huawei.com>
volcano-sh · Oct 18, 2022 · 1de3e11 · 1de3e11
1 parent 59e58c2
commit 1de3e11
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/pkg/scheduler/framework/arguments.go b/pkg/scheduler/framework/arguments.go
@@ -58,6 +58,10 @@ func (a Arguments) GetFloat64(ptr *float64, key string) {
 
 	value, ok := argv.(float64)
 	if !ok {
+		if intVal, ok := argv.(int); ok {
+			*ptr = float64(intVal)
+			return
+		}
 		klog.Warningf("Could not parse argument: %v for key %s to float64", argv, key)
 		return
 	}

diff --git a/pkg/scheduler/framework/arguments_test.go b/pkg/scheduler/framework/arguments_test.go
@@ -116,6 +116,15 @@ func TestArgumentsGetFloat64(t *testing.T) {
 			baseValue:   1.2,
 			expectValue: 1.2,
 		},
+		{
+			name: "int value",
+			arg: Arguments{
+				key1: 15,
+			},
+			key:         key1,
+			baseValue:   1.2,
+			expectValue: 15,
+		},
 	}
 
 	for index, c := range cases {