Skip to content

Commit

Permalink
Fix formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
plutasnyy committed Apr 15, 2024
1 parent 0b0d870 commit b23fc50
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions unstructured_inference/models/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,10 +24,12 @@
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},
Expand Down

0 comments on commit b23fc50

Please sign in to comment.