Skip to content

Commit

Permalink
treat bedrock as ingest
Browse files Browse the repository at this point in the history
  • Loading branch information
badGarnet committed Oct 18, 2023
1 parent 1c82b19 commit c148f05
Show file tree
Hide file tree
Showing 3 changed files with 1 addition and 3 deletions.
File renamed without changes.
File renamed without changes.
4 changes: 1 addition & 3 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,6 @@ def load_requirements(file_list: Optional[Union[str, List[str]]] = None) -> List


csv_reqs = load_requirements("requirements/extra-csv.in")
bedrock_reqs = load_requirements("requirements/extra-bedrock.in")
doc_reqs = load_requirements("requirements/extra-docx.in")
docx_reqs = load_requirements("requirements/extra-docx.in")
epub_reqs = load_requirements("requirements/extra-epub.in")
Expand All @@ -60,7 +59,6 @@ def load_requirements(file_list: Optional[Union[str, List[str]]] = None) -> List
all_doc_reqs = list(
set(
csv_reqs
+ bedrock_reqs
+ docx_reqs
+ epub_reqs
+ image_reqs
Expand Down Expand Up @@ -113,7 +111,6 @@ def load_requirements(file_list: Optional[Union[str, List[str]]] = None) -> List
# Document specific extra requirements
"all-docs": all_doc_reqs,
"csv": csv_reqs,
"bedrock": bedrock_reqs,
"doc": doc_reqs,
"docx": docx_reqs,
"epub": epub_reqs,
Expand Down Expand Up @@ -161,6 +158,7 @@ def load_requirements(file_list: Optional[Union[str, List[str]]] = None) -> List
"local-inference": all_doc_reqs,
"paddleocr": load_requirements("requirements/extra-paddleocr.in"),
"openai": load_requirements("requirements/ingest-openai.in"),
"bedrock": load_requirements("requirements/ingest-bedrock.in"),
},
package_dir={"unstructured": "unstructured"},
package_data={"unstructured": ["nlp/*.txt"]},
Expand Down

0 comments on commit c148f05

Please sign in to comment.