From 19189603e80ff486a4eda0d2fce43a110fa97733 Mon Sep 17 00:00:00 2001
From: omar <omarchamaalak@gmail.com>
Date: Tue, 20 Feb 2024 16:16:59 +0100
Subject: [PATCH] Flake8 Format fixes, Deleted obsolete options class

---
 main.py                            |  3 ++-
 options/options_text_generation.py | 25 -------------------------
 2 files changed, 2 insertions(+), 26 deletions(-)
 delete mode 100644 options/options_text_generation.py

diff --git a/main.py b/main.py
index ca230eb..7042ef2 100644
--- a/main.py
+++ b/main.py
@@ -4,7 +4,8 @@
 
 if __name__ == '__main__':
 
-    parser = argparse.ArgumentParser(description='Choose a Model type : TextConv or TxtToImg')
+    parser = argparse.ArgumentParser(description='Choose a Model type :'
+                                                 ' TextConv or TxtToImg')
     subparser = parser.add_subparsers(dest='option')
 
     conv = subparser.add_parser('TextConv')
diff --git a/options/options_text_generation.py b/options/options_text_generation.py
deleted file mode 100644
index a386070..0000000
--- a/options/options_text_generation.py
+++ /dev/null
@@ -1,25 +0,0 @@
-from options.options import Options, Devices
-
-
-class OptionsTextGeneration(Options):
-    """
-    Options for text-Generation models
-    """
-    prompt: str
-    max_length: int
-    temperature: float
-
-    def __init__(self, prompt: str, device: Devices, max_length: int, temperature: float):
-        """
-        Initializes the OptionsTextGeneration
-        :param device: The device to use generate prompt
-        :param prompt: The prompt to give to the model
-        :param max_length: The max length of the generated response
-        :param temperature: parameter used during the
-         sampling process to control the randomness of generated text
-         High temp : High randomness, Low temp : Low randomness
-        """
-        super().__init__(device)
-        self.prompt = prompt
-        self.max_length = max_length
-        self.temperature = temperature