Add omitempty tag to parallel_workers field (#603)

scheduler/pkg/agent/repository/mlserver/mlserver.go

@@ -64,7 +64,7 @@ type ModelSettings struct {
 	Outputs         []ModelMetadataTensors `json:"outputs,omitempty"`
 	Platform        string                 `json:"platform,omitempty"`
 	Versions        []string               `json:"versions,omitempty"`
-	ParallelWorkers int                    `json:"parallel_workers"`
+	ParallelWorkers int                    `json:"parallel_workers,omitempty"`
 	Implementation  string                 `json:"implementation,omitempty"`
 	Parameters      *ModelParameters       `json:"parameters,omitempty"`
 }

scheduler/pkg/agent/repository/mlserver/mlserver_test.go

@@ -730,3 +730,30 @@ func TestFindHighestVersionInPath(t *testing.T) {
 		})
 	}
 }
+
+func TestDefaultModelSettings(t *testing.T) {
+	g := NewGomegaWithT(t)
+
+	tests := []struct {
+		name          string
+		modelSettings *ModelSettings
+		expected      []byte
+	}{
+		{name: "omits all empty fields",
+			modelSettings: &ModelSettings{Name: "foo"},
+			expected:      []byte("{\"name\":\"foo\"}"),
+		},
+		{
+			name:          "adds empty parameters dict",
+			modelSettings: &ModelSettings{Name: "foo", Parameters: &ModelParameters{}},
+			expected:      []byte("{\"name\":\"foo\",\"parameters\":{}}"),
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			data, _ := json.Marshal(test.modelSettings)
+			g.Expect(data).To(Equal(test.expected))
+		})
+	}
+}