Make tidy

Unstructured-IO · Apr 15, 2024 · f68e5be · f68e5be
1 parent cd35ba7
commit f68e5be
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 9 deletions.
diff --git a/unstructured_inference/models/base.py b/unstructured_inference/models/base.py
@@ -24,12 +24,10 @@
 models: Dict[str, UnstructuredModel] = {}
 
 
-def get_default_model_mappings() -> (
-    Tuple[
-        Dict[str, Type[UnstructuredModel]],
-        Dict[str, dict | LazyDict],
-    ]
-):
+def get_default_model_mappings() -> Tuple[
+    Dict[str, Type[UnstructuredModel]],
+    Dict[str, dict | LazyDict],
+]:
     """default model mappings for models that are in `unstructured_inference` repo"""
     return {
         **{name: UnstructuredDetectronModel for name in DETECTRON2_MODEL_TYPES},

diff --git a/unstructured_inference/models/tables.py b/unstructured_inference/models/tables.py
@@ -27,7 +27,7 @@ class UnstructuredTableTransformerModel(UnstructuredModel):
     def __init__(self):
         pass
 
-    def predict(self, x: Image, ocr_tokens: Optional[List[Dict]] = None, result_format='html'):
+    def predict(self, x: Image, ocr_tokens: Optional[List[Dict]] = None, result_format="html"):
         """Predict table structure deferring to run_prediction with ocr tokens
 
         Note:
@@ -113,8 +113,9 @@ def run_prediction(
             prediction = prediction
         else:
             raise ValueError(
-                f'result_format {result_format} is not a valid format. '
-                f'Valid formats are: "html", "dataframe", "cells"')
+                f"result_format {result_format} is not a valid format. "
+                f'Valid formats are: "html", "dataframe", "cells"'
+            )
 
         return prediction