diff --git a/prompt2model/dataset_generator/prompt_based.py b/prompt2model/dataset_generator/prompt_based.py
index db2824446..354b12254 100644
--- 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):