ruff format + lint

Author: benjaminye

test_utils/test_config.py

@@ -6,10 +6,10 @@
     InferenceConfig,
     LoraConfig,
     ModelConfig,
+    QaConfig,
     SftArgs,
     TrainingArgs,
     TrainingConfig,
-    QaConfig
 )
 
 
@@ -72,6 +72,15 @@ def get_sample_config():
             train_test_split_seed=42,
         ),
         qa=QaConfig(
-            llm_tests=["jaccard_similarity", "dot_product", "rouge_score", "word_overlap", "verb_percent", "adjective_percent", "noun_percent", "summary_length"]
-        )
+            llm_tests=[
+                "jaccard_similarity",
+                "dot_product",
+                "rouge_score",
+                "word_overlap",
+                "verb_percent",
+                "adjective_percent",
+                "noun_percent",
+                "summary_length",
+            ]
+        ),
     )

tests/qa/test_generics.py

@@ -65,9 +65,7 @@ def test_save_test_results(mock_csv, mock_tests, mocker):
 
 def test_print_test_results(capfd, example_data):
     tests = [MockQaTest()]
-    test_suite = LLMTestSuite(
-        tests, example_data["Prompt"], example_data["Ground Truth"], example_data["Predicted"]
-    )
+    test_suite = LLMTestSuite(tests, example_data["Prompt"], example_data["Ground Truth"], example_data["Predicted"])
     test_suite.print_test_results()
     out, err = capfd.readouterr()