fix gcn scorer tests

Author: voorhs

tests/modules/scoring/test_gcn_scorer.py

@@ -45,7 +45,13 @@ def multiclass_dataset():
 
 def test_gcn_scorer_multilabel(multilabel_dataset):
     torch.manual_seed(42)
-    scorer = GCNScorer(embedder_config=get_test_embedder_config(), num_train_epochs=1, batch_size=2, seed=42)
+    scorer = GCNScorer(
+        embedder_config=get_test_embedder_config(),
+        label_embedder_config=get_test_embedder_config(),
+        num_train_epochs=1,
+        batch_size=2,
+        seed=42,
+    )
     train_utterances = multilabel_dataset["train"]["utterance"]
     train_labels = multilabel_dataset["train"]["label"]
     descriptions = [intent.name for intent in multilabel_dataset.intents]
@@ -60,7 +66,13 @@ def test_gcn_scorer_multilabel(multilabel_dataset):
 
 def test_gcn_scorer_multiclass(multiclass_dataset):
     torch.manual_seed(42)
-    scorer = GCNScorer(embedder_config=get_test_embedder_config(), num_train_epochs=1, batch_size=2, seed=42)
+    scorer = GCNScorer(
+        embedder_config=get_test_embedder_config(),
+        label_embedder_config=get_test_embedder_config(),
+        num_train_epochs=1,
+        batch_size=2,
+        seed=42,
+    )
     train_utterances = multiclass_dataset["train"]["utterance"]
     train_labels = multiclass_dataset["train"]["label"]
     descriptions = [intent.name for intent in multiclass_dataset.intents]
@@ -76,7 +88,13 @@ def test_gcn_scorer_multiclass(multiclass_dataset):
 
 def test_gcn_scorer_dump_load(tmp_path, multilabel_dataset):
     torch.manual_seed(42)
-    scorer = GCNScorer(embedder_config=get_test_embedder_config(), num_train_epochs=1, batch_size=2, seed=42)
+    scorer = GCNScorer(
+        embedder_config=get_test_embedder_config(),
+        label_embedder_config=get_test_embedder_config(),
+        num_train_epochs=1,
+        batch_size=2,
+        seed=42,
+    )
     train_utterances = multilabel_dataset["train"]["utterance"]
     train_labels = multilabel_dataset["train"]["label"]
     descriptions = [intent.name for intent in multilabel_dataset.intents]