Skip to content

Commit 904a815

Browse files
Remove option from TEST CLI
1 parent fa182e4 commit 904a815

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

boudams/cli.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -342,16 +342,16 @@ def train(
342342
@cli.command("test")
343343
@click.argument("test_path", type=click.Path(dir_okay=False, file_okay=True, exists=True))
344344
@click.argument("models", nargs=-1, type=click.Path(dir_okay=False, file_okay=True, exists=True))
345-
@click.option("--csv_file", default=None, type=click.File(mode="w"), help="CSV target")
345+
#@click.option("--csv_file", default=None, type=click.File(mode="w"), help="CSV target")
346346
@click.option("--batch_size", type=int, default=32, help="Size of batches")
347347
@click.option("--device", default="cpu", help="Device to use for the network (cuda, cpu, etc.)")
348348
@click.option("--debug", default=False, is_flag=True)
349-
@click.option("--verbose", default=False, is_flag=True, help="Print classification report")
349+
#@click.option("--verbose", default=False, is_flag=True, help="Print classification report")
350350
@click.option("--workers", default=1, type=int, help="Number of workers to use to load data")
351351
@click.option("--avg", default="macro", type=click.Choice(["micro", "macro"]), help="Type of avering method to use on "
352352
"metrics")
353-
def test(test_path, models, csv_file, batch_size, device, debug, verbose, workers: int, avg: str):
354-
""" Train one or more models according to [CONFIG_FILES] JSON configurations"""
353+
def test(test_path, models, batch_size, device, debug, workers: int, avg: str):
354+
""" Test one or many [MODELS] on the file at TEST_PATH """
355355
if debug:
356356
logger.setLevel(logging.DEBUG)
357357
else:

0 commit comments

Comments
 (0)