From 4f100318f499cff48b62555ca7d96ad97d7cb9be Mon Sep 17 00:00:00 2001
From: YuryYakhno <52328128+YuryYakhno@users.noreply.github.com>
Date: Fri, 3 Nov 2023 21:06:12 +0300
Subject: [PATCH] Add explicit error if empty batch received (#2115)

* Add explicit error if empty batch received

* Move error check to cover all empty iterables
---
 src/accelerate/utils/operations.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/accelerate/utils/operations.py b/src/accelerate/utils/operations.py
index ee91b60c4eb..267f7809ef0 100644
--- a/src/accelerate/utils/operations.py
+++ b/src/accelerate/utils/operations.py
@@ -231,6 +231,9 @@ def find_batch_size(data):
     Returns:
         `int`: The batch size.
     """
+    if isinstance(data, (tuple, list, Mapping)) and (len(data) == 0):
+        raise ValueError(f"Cannot find the batch size from empty {type(data)}.")
+
     if isinstance(data, (tuple, list)):
         return find_batch_size(data[0])
     elif isinstance(data, Mapping):