From d1194abc745ac275e9523fa53f0c86c3cb3b4c7e Mon Sep 17 00:00:00 2001 From: Aashiq Muhamed Date: Wed, 12 Apr 2023 21:07:01 -0700 Subject: [PATCH] Bug fix in setattr --- src/accelerate/data_loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/accelerate/data_loader.py b/src/accelerate/data_loader.py index d06ffbf69fc..7f8d12f9812 100644 --- a/src/accelerate/data_loader.py +++ b/src/accelerate/data_loader.py @@ -407,7 +407,7 @@ def total_batch_size(self): @property def total_dataset_length(self): - if hasattr("total_length", self.dataset): + if hasattr(self.dataset, "total_length"): return self.dataset.total_length else: return len(self.dataset)