From 8a2219b9d36f91f0d7715398b07ba46ee5cd7556 Mon Sep 17 00:00:00 2001
From: Yahya Badran <yahya.badran@gmail.com>
Date: Tue, 2 Jun 2026 00:47:40 +0200
Subject: [PATCH] fix: handle empty inputs in OrientationPredictor

---
 doctr/models/classification/predictor/pytorch.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/doctr/models/classification/predictor/pytorch.py b/doctr/models/classification/predictor/pytorch.py
index af59b4a8c7..280743af4d 100644
--- a/doctr/models/classification/predictor/pytorch.py
+++ b/doctr/models/classification/predictor/pytorch.py
@@ -37,6 +37,9 @@ def forward(
         self,
         inputs: list[np.ndarray],
     ) -> list[list[int] | list[float]]:
+        if len(inputs) == 0:
+            return [[], [], []]
+
         # Dimension check
         if any(input.ndim != 3 for input in inputs):
             raise ValueError("incorrect input shape: all inputs are expected to be multi-channel 2D images.")