fix possible key error

donglihe-hub · donglihe-hub · commit 86e47bee90b3 · 2024-01-22T17:52:26.000+04:00
diff --git a/libmultilabel/nn/data_utils.py b/libmultilabel/nn/data_utils.py
@@ -243,7 +243,7 @@ def load_datasets(
             test_data, is_test=True, tokenize_text=tokenize_text, remove_no_label_data=remove_no_label_data
         )
 
-    if merge_train_val:
+    if merge_train_val and "val" in datasets:
         datasets["train"] = datasets["train"] + datasets["val"]
         for i in range(len(datasets["train"])):
             datasets["train"][i]["index"] = i

Original file line number	Diff line number	Diff line change
`@@ -243,7 +243,7 @@ def load_datasets(`
`243`	`243`	`test_data, is_test=True, tokenize_text=tokenize_text, remove_no_label_data=remove_no_label_data`
`244`	`244`	`)`
`245`	`245`
`246`		`- if merge_train_val:`
	`246`	`+ if merge_train_val and "val" in datasets:`
`247`	`247`	`datasets["train"] = datasets["train"] + datasets["val"]`
`248`	`248`	`for i in range(len(datasets["train"])):`
`249`	`249`	`datasets["train"][i]["index"] = i`