Update prompt_based.py

neulab · Dec 5, 2023 · 0802cdd · 0802cdd
1 parent 6e36a03
commit 0802cdd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/prompt2model/dataset_generator/prompt_based.py b/prompt2model/dataset_generator/prompt_based.py
@@ -41,16 +41,16 @@ class Example:
     def __eq__(self, other) -> bool:
         """Example equality."""
         return (
-            self.input_col == other.input_col 
-            and self.output_col == other.output_col 
+            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 
+            self.input_col < other.input_col
+            or self.output_col < other.output_col
             or self.explain_col < other.explain_col
         ) # noqa E501