feat: update CLI tools for dataloader module

- Update valid modules list in test command: data → dataloader
- Update module display name in info command: Data → DataLoader
- Update CLI references to use new module name
- All CLI commands now recognize 'dataloader' instead of 'data'

Ensures CLI tools work seamlessly with the renamed module.
This commit is contained in:
Vijay Janapa Reddi
2025-07-11 18:59:19 -04:00
parent 8331f93438
commit 8b6eabe810
2 changed files with 2 additions and 2 deletions

View File

@@ -78,7 +78,7 @@ class InfoCommand(BaseCommand):
("Networks", "neural network architectures", self.check_networks_status),
("MLP", "multi-layer perceptron (manual)", self.check_mlp_status),
("CNN", "convolutional networks (basic)", self.check_cnn_status),
("Data", "data loading pipeline", self.check_data_status),
("DataLoader", "data loading pipeline", self.check_data_status),
("Training", "autograd engine & optimization", self.check_training_status),
("Profiling", "performance profiling", self.check_profiling_status),
("Compression", "model compression", self.check_compression_status),

View File

@@ -32,7 +32,7 @@ class TestCommand(BaseCommand):
def run(self, args: Namespace) -> int:
console = self.console
valid_modules = ["setup", "tensor", "activations", "layers", "networks", "cnn", "data", "training",
valid_modules = ["setup", "tensor", "activations", "layers", "networks", "cnn", "dataloader", "training",
"profiling", "compression", "kernels", "benchmarking", "mlops"]
if args.all: