From 78f56a6d12af493f7a5e94b05ce8c3fd2e1deefb Mon Sep 17 00:00:00 2001 From: ianbulovic Date: Fri, 10 Apr 2026 12:19:27 -0400 Subject: [PATCH] add negative options for CLI flags that default to true --- src/cnlpt/_cli/train.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/src/cnlpt/_cli/train.py b/src/cnlpt/_cli/train.py index 44dc1544..6d2f7738 100644 --- a/src/cnlpt/_cli/train.py +++ b/src/cnlpt/_cli/train.py @@ -337,11 +337,18 @@ def transformers_arg_option(field_name: str, *args, **kwargs): FreezeEncoderArg = Annotated[float, training_arg_option("freeze_encoder")] BiasFitArg = Annotated[bool, training_arg_option("bias_fit")] EvalsPerEpochArg = Annotated[int, training_arg_option("evals_per_epoch")] -RichDisplayArg = Annotated[bool, training_arg_option("rich_display")] +RichDisplayArg = Annotated[ + bool, training_arg_option("rich_display", "--rich_display/--no_rich_display") +] LoggingStrategyArg = Annotated[ IntervalStrategy, training_arg_option("logging_strategy") ] -LoggingFirstStepArg = Annotated[bool, training_arg_option("logging_first_step")] +LoggingFirstStepArg = Annotated[ + bool, + training_arg_option( + "logging_first_step", "--logging_first_step/--no_logging_first_step" + ), +] CacheDirArg = Annotated[Union[str, None], training_arg_option("cache_dir")] MetricForBestModelArg = Annotated[str, training_arg_option("metric_for_best_model")]