From 0c60dfb6c07b3757c314fc0671c5be4bc247a55c Mon Sep 17 00:00:00 2001 From: Vijay Janapa Reddi Date: Fri, 11 Jul 2025 21:56:19 -0400 Subject: [PATCH] fix: correct DataLoader status check in CLI MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - Fix CLI test to use Tensor objects instead of raw integers - DataLoader now correctly shows as ✅ Implemented in status - Test creates proper Tensor data for DataLoader compatibility --- tito/commands/info.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tito/commands/info.py b/tito/commands/info.py index 2ee4279a..d459cf9a 100644 --- a/tito/commands/info.py +++ b/tito/commands/info.py @@ -203,8 +203,9 @@ class InfoCommand(BaseCommand): def check_data_status(self): try: from tinytorch.core.dataloader import DataLoader + from tinytorch.core.tensor import Tensor import numpy as np - data = [(np.random.randn(3,32,32), 0) for _ in range(10)] + data = [(Tensor(np.random.randn(3,32,32)), Tensor(np.array(i % 10))) for i in range(10)] loader = DataLoader(data, batch_size=2, shuffle=True) _ = next(iter(loader)) return "✅ Implemented"