From 89dc7094c8c27c204538776515162da76d1ad775 Mon Sep 17 00:00:00 2001
From: Kyle Sayers <kylesayrs@gmail.com>
Date: Mon, 10 Nov 2025 16:20:01 +0000
Subject: [PATCH] fix

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
---
 src/llmcompressor/datasets/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/llmcompressor/datasets/utils.py b/src/llmcompressor/datasets/utils.py
index 8aef67fcd..673968d40 100644
--- a/src/llmcompressor/datasets/utils.py
+++ b/src/llmcompressor/datasets/utils.py
@@ -129,7 +129,7 @@ def format_calibration_data(
     num_calibration_samples: int | None = None,
     do_shuffle: bool = True,
     collate_fn: Callable = default_data_collator,
-) -> list[torch.Tensor]:
+) -> DataLoader:
     """
     Creates a dataloader out of the calibration dataset split, trimming it to
     the desired number of calibration samples