From 2ae8b7589e1d859196a1692527c28c139e4e0825 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= <rmontanana@gmail.com>
Date: Wed, 4 May 2022 19:46:14 +0200
Subject: [PATCH] Fix be_main args issue

---
 benchmark/Arguments.py       | 11 ++++++++++-
 benchmark/scripts/be_main.py |  7 +++----
 2 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/benchmark/Arguments.py b/benchmark/Arguments.py
index 333bcd6..3325d62 100644
--- a/benchmark/Arguments.py
+++ b/benchmark/Arguments.py
@@ -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"),
                 {
diff --git a/benchmark/scripts/be_main.py b/benchmark/scripts/be_main.py
index 95cb060..34ccc95 100755
--- a/benchmark/scripts/be_main.py
+++ b/benchmark/scripts/be_main.py
@@ -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)