Update prompt_based.py

neulab · Dec 5, 2023 · 7b6cd50 · 7b6cd50
1 parent 49a41d9
commit 7b6cd50
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/prompt2model/dataset_generator/prompt_based.py b/prompt2model/dataset_generator/prompt_based.py
@@ -40,15 +40,11 @@ class Example:
 
     def __eq__(self, other) -> bool:
         """Example equality."""
-        return (self.input_col == other.input_col and
-                self.output_col == other.output_col and
-                self.explain_col == other.explain_col)
+        return (self.input_col == other.input_col and self.output_col == other.output_col and self.explain_col == other.explain_col) # noqa E501
 
     def __lt__(self, other) -> bool:
         """Example less than."""
-        return (self.input_col < other.input_col
-                or self.output_col < other.output_col
-                or self.explain_col < other.explain_col)
+        return (self.input_col < other.input_col or self.output_col < other.output_col or self.explain_col < other.explain_col) # noqa E501
 
 
 class PromptBasedDatasetGenerator(DatasetGenerator):