Don't instantiate input format by default, create it in options funct…

…ion if nil
h2oai · Jan 22, 2024 · 0095e4d · 0095e4d
1 parent 2a94357
commit 0095e4d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/task_types.go b/task_types.go
@@ -85,7 +85,7 @@ func defaultTaskIngestionSpec() *TaskIngestionSpec {
 			IOConfig: &IOConfig{
 				Type:        "index_parallel",
 				InputSource: &InputSource{},
-				InputFormat: &InputFormat{},
+				// InputFormat: &InputFormat{},
 			},
 			TuningConfig: &TuningConfig{
 				Type: "index_parallel",
@@ -176,6 +176,10 @@ func SetTaskIOConfigType(typ string) TaskIngestionSpecOptions {
 // SetTaskInputFormat configures input format for the task based ingestion.
 func SetTaskInputFormat(typ string, findColumnsHeader string, columns []string) TaskIngestionSpecOptions {
 	return func(spec *TaskIngestionSpec) {
+		if spec.Spec.IOConfig.InputFormat == nil {
+			spec.Spec.IOConfig.InputFormat = &InputFormat{}
+		}
+
 		spec.Spec.IOConfig.InputFormat.Type = typ
 		spec.Spec.IOConfig.InputFormat.FindColumnsFromHeader = findColumnsHeader
 		spec.Spec.IOConfig.InputFormat.Columns = columns