lint

Author: Samoed

tests/modules/scoring/test_gcn_scorer.py

@@ -47,7 +47,13 @@ def multiclass_dataset():
 
 def test_gcn_scorer_multilabel(multilabel_dataset):
     torch.manual_seed(42)
-    scorer = GCNScorer(embedder_config=_embedder_config, label_embedder_config=_embedder_config, num_train_epochs=1, batch_size=2, seed=42)
+    scorer = GCNScorer(
+        embedder_config=_embedder_config,
+        label_embedder_config=_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]
@@ -62,7 +68,13 @@ def test_gcn_scorer_multilabel(multilabel_dataset):
 
 def test_gcn_scorer_multiclass(multiclass_dataset):
     torch.manual_seed(42)
-    scorer = GCNScorer(embedder_config=_embedder_config, label_embedder_config=_embedder_config, num_train_epochs=1, batch_size=2, seed=42)
+    scorer = GCNScorer(
+        embedder_config=_embedder_config,
+        label_embedder_config=_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]
@@ -78,7 +90,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=_embedder_config, label_embedder_config=_embedder_config, num_train_epochs=1, batch_size=2, seed=42)
+    scorer = GCNScorer(
+        embedder_config=_embedder_config,
+        label_embedder_config=_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]