1

src/rev_ai/models/asynchronous/summarization_formatting_options.py

@@ -1,7 +1,7 @@
 from enum import Enum
-"""Summarization formatting options."""
 
 
 class SummarizationFormattingOptions(str, Enum):
+    """Summarization formatting options."""
     PARAGRAPH = "paragraph"
     BULLETS = "bullets"

src/rev_ai/models/asynchronous/summarization_job_status.py

@@ -1,10 +1,9 @@
 # -*- coding: utf-8 -*-
-"""Enum for Summarization Job statuses"""
-
 from enum import Enum
 
 
 class SummarizationJobStatus(str, Enum):
+    """Enum for Summarization Job statuses"""
     IN_PROGRESS = "in_progress"
     FAILED = "failed"
     COMPLETED = "completed"

src/rev_ai/models/asynchronous/summarization_options.py

@@ -10,8 +10,7 @@ def __init__(
             self,
             prompt: str = None,
             model: NlpModel = None,
-            formatting_type: SummarizationFormattingOptions=None
-    ):
+            formatting_type: SummarizationFormattingOptions = None):
         self.prompt = prompt
         self.model = model
         self.type = formatting_type
@@ -40,8 +39,7 @@ def __init__(
             formatting_type: SummarizationFormattingOptions = None,
             status: SummarizationJobStatus = None,
             completed_on: str = None,
-            failure: str = None
-    ):
+            failure: str = None):
         super().__init__(prompt, model, formatting_type)
         self.status = status
         self.completed_on = completed_on

src/rev_ai/models/asynchronous/summary.py

@@ -7,8 +7,7 @@ class Summary:
     def __init__(
             self,
             summary: str,
-            bullet_points: List[str],
-    ):
+            bullet_points: List[str]):
         self.summary = summary
         self.bullet_points = bullet_points
 

src/rev_ai/models/asynchronous/translation_job_status.py

@@ -1,10 +1,9 @@
 # -*- coding: utf-8 -*-
-"""Enum for Translation Job statuses"""
-
 from enum import Enum
 
 
 class TranslationJobStatus(str, Enum):
+    """Enum for Translation Job statuses"""
     IN_PROGRESS = "in_progress"
     FAILED = "failed"
     COMPLETED = "completed"

src/rev_ai/models/asynchronous/translation_language_options.py

@@ -1,15 +1,13 @@
 from .summarization_job_status import SummarizationJobStatus
 from ..nlp_model import NlpModel
 
-"""Translation language request options."""
-
 
 class TranslationLanguageOptions:
+    """Translation language request options."""
     def __init__(
             self,
             language: str = None,
-            model: NlpModel = None
-    ):
+            model: NlpModel = None):
         self.language = language
         self.model = model
 
@@ -25,18 +23,14 @@ def to_dict(self):
         return dict_result
 
 
-"""Translation language options."""
-
-
 class TranslationLanguage(TranslationLanguageOptions):
+    """Translation language options."""
     def __init__(
             self,
             language: str = None,
             model: NlpModel = None,
             status: SummarizationJobStatus = None,
-            failure: str = None
-
-    ):
+            failure: str = None):
         super().__init__(language, model)
         self.status = status
         self.failure = failure

src/rev_ai/models/asynchronous/translation_options.py

@@ -1,34 +1,28 @@
 from typing import List, Dict
 from .translation_language_options import TranslationLanguageOptions, TranslationLanguage
 
-"""Translation request options."""
-
 
 class TranslationOptions:
+    """Translation request options."""
     def __init__(
             self,
-            target_languages: List[TranslationLanguageOptions]
-    ):
+            target_languages: List[TranslationLanguageOptions]):
         self.target_languages = target_languages
 
     def to_dict(self):
         """Returns the raw form of the url data object as the api
         expects them"""
-        dict_result = {}
-        dict_result["target_languages"] = [tl.to_dict() for tl in self.target_languages]
+        dict_result = {"target_languages": [tl.to_dict() for tl in self.target_languages]}
 
         return dict_result
 
 
-"""Translation options."""
-
-
 class Translation(TranslationOptions):
+    """Translation options."""
     def __init__(
             self,
             target_languages: List[TranslationLanguageOptions],
-            completed_on: str = None,
-    ):
+            completed_on: str = None):
         super().__init__(target_languages)
         self.completed_on = completed_on