Doctorado-ML/benchmark
1
0
Fork 0
mirror of https://github.com/Doctorado-ML/benchmark.git synced 2025-08-18 17:05:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor Arguments to be child of ArgumentParser

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2022-11-19 21:25:50 +01:00
parent 07172b91c5
commit 146304f4b5
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
benchmark/Arguments.py
View File

@@ -48,9 +48,9 @@ class EnvDefault(argparse.Action):
setattr(namespace, self.dest, values) setattr(namespace, self.dest, values)
class Arguments: class Arguments(argparse.ArgumentParser):
def __init__(self): def __init__(self):
self.ap = argparse.ArgumentParser() super().__init__()
models_data = Models.define_models(random_state=0) models_data = Models.define_models(random_state=0)
self._overrides = {} self._overrides = {}
self.parameters = { self.parameters = {
@@ -303,7 +303,7 @@ class Arguments:
if "overrides" in kwargs: if "overrides" in kwargs:
self._overrides[names[0]] = (kwargs["overrides"], kwargs["const"]) self._overrides[names[0]] = (kwargs["overrides"], kwargs["const"])
del kwargs["overrides"] del kwargs["overrides"]
self.ap.add_argument( self.add_argument(
*names, *names,
**{**parameters, **kwargs}, **{**parameters, **kwargs},
) )
@@ -315,4 +315,4 @@ class Arguments:
args = sys.argv[1:] args = sys.argv[1:]
if key in args: if key in args:
args.extend((f"--{dest_key}", value)) args.extend((f"--{dest_key}", value))
return self.ap.parse_args(args) return super().parse_args(args)