diff --git a/lib/sycamore/sycamore/transforms/detr_partitioner.py b/lib/sycamore/sycamore/transforms/detr_partitioner.py index 234499953..2b367e0e6 100644 --- a/lib/sycamore/sycamore/transforms/detr_partitioner.py +++ b/lib/sycamore/sycamore/transforms/detr_partitioner.py @@ -38,12 +38,6 @@ _VERSION = "0.2024.07.24" -def _batchify(iterable, n=1): - length = len(iterable) - for i in range(0, length, n): - yield iterable[i : min(i + n, length)] - - ARYN_DETR_MODEL = "Aryn/deformable-detr-DocLayNet" DEFAULT_ARYN_PARTITIONER_ADDRESS = "https://api.aryn.cloud/v1/document/partition" _TEN_MINUTES = 600