fix: correct DataLoader status check in CLI

- 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
This commit is contained in:
Vijay Janapa Reddi
2025-07-11 21:56:19 -04:00
parent a51c8048cb
commit 0c60dfb6c0

View File

@@ -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"