Skip to content

Commit 6eb94de

Browse files
committed
fix: fixed unpacking issues after rebase
1 parent f2582a0 commit 6eb94de

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/unstructured_client/_hooks/custom/split_pdf_hook.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -374,7 +374,7 @@ def _get_pdf_pages(
374374
split_size: int = 1,
375375
page_start: int = 1,
376376
page_end: Optional[int] = None
377-
) -> Generator[Tuple[io.BytesIO, int, int], None, None]:
377+
) -> Generator[Tuple[io.BytesIO, int], None, None]:
378378
"""Reads given bytes of a pdf file and split it into n file-like objects, each
379379
with `split_size` pages.
380380
@@ -410,10 +410,10 @@ def _get_pdf_pages(
410410
new_pdf.add_page(page)
411411
with open(tempdir_path / f"chunk_{chunk_no}.pdf", "wb") as pdf_chunk:
412412
new_pdf.write(pdf_chunk)
413-
pdf_chunks.append((pdf_chunk.name, offset, offset_end))
413+
pdf_chunks.append((pdf_chunk.name, offset))
414414
offset += split_size
415415

416-
for pdf_chunk_filename, offset, offset_end in pdf_chunks:
416+
for pdf_chunk_filename, offset in pdf_chunks:
417417
pdf_chunk_file = None
418418
try:
419419
pdf_chunk_file = open( # pylint: disable=consider-using-with
@@ -423,7 +423,7 @@ def _get_pdf_pages(
423423
except Exception: # pylint: disable=broad-except
424424
if pdf_chunk_file and not pdf_chunk_file.closed:
425425
pdf_chunk_file.close()
426-
yield pdf_chunk_file, offset, offset_end
426+
yield pdf_chunk_file, offset
427427

428428
def _await_elements(
429429
self, operation_id: str) -> Optional[list]:

0 commit comments

Comments
 (0)