From 8b6eabe81071fd8e92707dcc4b080c2f2b201b35 Mon Sep 17 00:00:00 2001 From: Vijay Janapa Reddi Date: Fri, 11 Jul 2025 18:59:19 -0400 Subject: [PATCH] feat: update CLI tools for dataloader module MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 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. --- tito/commands/info.py | 2 +- tito/commands/test.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tito/commands/info.py b/tito/commands/info.py index 21f25ee7..2ee4279a 100644 --- a/tito/commands/info.py +++ b/tito/commands/info.py @@ -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), diff --git a/tito/commands/test.py b/tito/commands/test.py index 14d05612..51da77ec 100644 --- a/tito/commands/test.py +++ b/tito/commands/test.py @@ -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: