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?",
},
],
"platform": [],
"platform": [
("-P", "--platform"),
{
"action": EnvDefault,
"envvar": "platform",
"type": str,
"required": True,
"help": "Platform where the test is run",
},
],
"quiet": [
("-q", "--quiet"),
{

View File

@@ -2,7 +2,7 @@
import os
from benchmark.Experiments import Experiment, Datasets
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
"""
@@ -28,15 +28,14 @@ def main():
grid_paramfile=args.grid_paramfile,
progress_bar=not args.quiet,
platform=args.platform,
title=args.experiment_title,
folds=args.folds,
title=args.title,
folds=args.n_folds,
)
job.do_experiment()
if report:
result_file = job.get_output_file()
report = Report(result_file)
report.report()
if args.dataset is not None:
print(f"Partial result file removed: {result_file}")
os.remove(result_file)