Source code for czbenchmarks.cli.cli_list

import argparse
import sys

from czbenchmarks.datasets import utils as dataset_utils
from czbenchmarks.models import utils as model_utils
from czbenchmarks.tasks import utils as task_utils


[docs] def add_arguments(parser: argparse.ArgumentParser) -> None: """ Add list command arguments to the parser. """ parser.add_argument( "list_type", choices=["datasets", "models", "tasks"], help="List available datasets, models, or tasks.", )
[docs] def main(args: argparse.Namespace) -> None: """ List available datasets, models, or tasks. """ if args.list_type == "datasets": sys.stdout.write(" ".join(dataset_utils.list_available_datasets())) elif args.list_type == "models": sys.stdout.write(" ".join(model_utils.list_available_models())) elif args.list_type == "tasks": sys.stdout.write(" ".join(task_utils.TASK_NAMES)) sys.stdout.write("\n")