Merge branch 'develop' into intelai/fix-gha

intel-staging · Nov 17, 2023 · 8c771c6 · 8c771c6
2 parents 5e83bb1 + 28e9044
commit 8c771c6
Showing 1 changed file with 5 additions and 12 deletions.
diff --git a/tests/evaluation_suite_glue_ci.py b/tests/evaluation_suite_glue_ci.py
@@ -3,7 +3,6 @@
 
 
 class Suite(evaluate.EvaluationSuite):
-
     def __init__(self, name):
         super().__init__(name)
 
@@ -19,11 +18,8 @@ def __init__(self, name):
                     "input_column": "sentence",
                     "label_column": "label",
                     "config_name": "sst2",
-                    "label_mapping": {
-                        "NEGATIVE": 0.0,
-                        "POSITIVE": 1.0
-                    }
-                }
+                    "label_mapping": {"NEGATIVE": 0.0, "POSITIVE": 1.0},
+                },
             ),
             SubTask(
                 task_type="text-classification",
@@ -36,10 +32,7 @@ def __init__(self, name):
                     "second_input_column": "question2",
                     "label_column": "label",
                     "config_name": "qqp",
-                    "label_mapping": {
-                        "NEGATIVE": 0.0,
-                        "POSITIVE": 1.0
-                    }
-                }
-            )
+                    "label_mapping": {"NEGATIVE": 0.0, "POSITIVE": 1.0},
+                },
+            ),
         ]