From 7b6cd50cb792ee5ebe95ede767f0924a81f5b571 Mon Sep 17 00:00:00 2001 From: VanyaBK <37258663+VanyaBK@users.noreply.github.com> Date: Mon, 4 Dec 2023 21:53:45 -0500 Subject: [PATCH] Update prompt_based.py --- prompt2model/dataset_generator/prompt_based.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) 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):