Fix be_main args issue

This commit is contained in:
2022-05-04 19:46:14 +02:00
parent 791991c929
commit 2ae8b7589e
2 changed files with 13 additions and 5 deletions

View File

@@ -206,7 +206,16 @@ class Arguments:
"help": "Use best hyperparams file?", "help": "Use best hyperparams file?",
}, },
], ],
"platform": [], "platform": [
("-P", "--platform"),
{
"action": EnvDefault,
"envvar": "platform",
"type": str,
"required": True,
"help": "Platform where the test is run",
},
],
"quiet": [ "quiet": [
("-q", "--quiet"), ("-q", "--quiet"),
{ {

View File

@@ -2,7 +2,7 @@
import os import os
from benchmark.Experiments import Experiment, Datasets from benchmark.Experiments import Experiment, Datasets
from benchmark.Results import Report from benchmark.Results import Report
from Arguments import Arguments from benchmark.Arguments import Arguments
"""Do experiment and build result file, optionally print report with results """Do experiment and build result file, optionally print report with results
""" """
@@ -28,15 +28,14 @@ def main():
grid_paramfile=args.grid_paramfile, grid_paramfile=args.grid_paramfile,
progress_bar=not args.quiet, progress_bar=not args.quiet,
platform=args.platform, platform=args.platform,
title=args.experiment_title, title=args.title,
folds=args.folds, folds=args.n_folds,
) )
job.do_experiment() job.do_experiment()
if report: if report:
result_file = job.get_output_file() result_file = job.get_output_file()
report = Report(result_file) report = Report(result_file)
report.report() report.report()
if args.dataset is not None: if args.dataset is not None:
print(f"Partial result file removed: {result_file}") print(f"Partial result file removed: {result_file}")
os.remove(result_file) os.remove(result_file)