Merge branch 'dev' into minhtrung23fix-pylint

minhtrung23 · Sep 10, 2024 · 40b4564 · 40b4564
2 parents ecc972a + cdb6ae0
commit 40b4564
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 3 deletions.
diff --git a/src/melt/__main__.py b/src/melt/__main__.py
@@ -11,7 +11,6 @@
 )
 logger = logging.getLogger("nlp_utils")
 
-
 def download_nltk_resources() -> NoReturn:
     """Download the necessary NLTK resources.
 

diff --git a/src/melt/tools/pipelines/metric_pipelines.py b/src/melt/tools/pipelines/metric_pipelines.py
@@ -19,7 +19,10 @@ def __init__(self):
             "question-answering": [QAMetric, BiasMetric, ToxicityMetric],
             "summarization": [SummaryMetric, BiasMetric, ToxicityMetric],
             "translation": [TranslationMetric, BiasMetric, ToxicityMetric],
-            "knowledge-mtpchoice": [QAMetric, BiasMetric, ToxicityMetric],
+            "knowledge-mtpchoice": [
+                TextClassificationMetric,
+                CalibrationMetric,
+            ],
             "knowledge-openended": [QAMetric, BiasMetric, ToxicityMetric],
             "toxicity-detection": [
                 TextClassificationMetric,

diff --git a/tests/test_execution.py b/tests/test_execution.py
@@ -1,6 +1,7 @@
 import subprocess
 import unittest
 
+
 class TestTasks(unittest.TestCase):
     """
     Unit tests for various tasks using the melt command-line tool.

diff --git a/tests/test_wrapper.py b/tests/test_wrapper.py
@@ -1,6 +1,7 @@
 import subprocess
 import unittest
 
+
 class TestWrapper(unittest.TestCase):
     """
     Unit tests for various wrappers used with the melt command-line tool.
@@ -79,5 +80,6 @@ def test_wrapper_vllm(self):
         """Test vllm wrapper."""
         self.run_melt_command("zalo_e2eqa", "vllm")
 
-if __name__ == '__main__':
+
+if __name__ == "__main__":
     unittest.main()