Skip to content
Open
Show file tree
Hide file tree
Changes from 4 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion indico/queries/datasets.py
Original file line number Diff line number Diff line change
Expand Up @@ -206,7 +206,7 @@ class CreateDataset(RequestChain["Dataset"]):
from_local_images (bool, optional): Flag whether files are local images or not. Defaults to False.
image_filename_col (str, optional): Image filename column. Defaults to 'filename'.
batch_size (int, optional): Size of file batch to upload at a time. Defaults to 20.
ocr_engine (OcrEngine, optional): Specify an OCR engine [OMNIPAGE, READAPI, READAPI_V2, READAPI_TABLES_V1]. Defaults to None.
ocr_engine (OcrEngine, optional): Specify an OCR engine [OMNIPAGE, READAPI, READAPI_V2, READAPI_TABLES_V1, READAPI_TABLES_V2]. Defaults to None.
omnipage_ocr_options (OmnipageOcrOptionsInput, optional): If using Omnipage, specify Omnipage OCR options. Defaults to None.
read_api_ocr_options: (ReadApiOcrOptionsInput, optional): If using ReadAPI, specify ReadAPI OCR options. Defaults to None.
request_interval (int or float, optional): The maximum time in between retry calls when waiting. Defaults to 5 seconds.
Expand Down
6 changes: 6 additions & 0 deletions indico/types/dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -84,6 +84,7 @@ class OcrEngine(Enum):
READAPI = 1
READAPI_V2 = 2
READAPI_TABLES_V1 = 3
READAPI_TABLES_V2 = 4


class OmnipageOcrOptionsInput(BaseType):
Expand Down Expand Up @@ -155,6 +156,11 @@ class EmailOptions(BaseType):
include_sections: Optional[IncludeSections]
unpack: Optional[bool]
preserve_body_whitespace: Optional[bool]
embed_inline_images: Optional[bool]
html_renderer_version: Optional[int]
msg_converter_version: Optional[int]
header_version: Optional[int]
sanitize_css: Optional[bool]


class OcrOptionsInput:
Expand Down
5 changes: 5 additions & 0 deletions tests/integration/queries/test_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -439,6 +439,11 @@ def test_create_with_email_options_readapi(indico):
email_config = {
"include_sections": {"header": True, "body": True, "attachments": True},
"unpack": True,
"embed_inline_images": True,
"html_renderer_version": 1,
"msg_converter_version": 1,
"header_version": 1,
"sanitize_css": False
}
dataset = client.call(
CreateEmptyDataset(
Expand Down