Skip to content

Commit

Permalink
[src] Fix argparse key based on Python version (#628) (#657)
Browse files Browse the repository at this point in the history
  • Loading branch information
jbartolewska committed Feb 22, 2023
1 parent 9dcf6ba commit bd3caa3
Showing 1 changed file with 9 additions and 2 deletions.
11 changes: 9 additions & 2 deletions asteroid/utils/parser_utils.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,6 @@
import argparse
import platform
from packaging.version import parse as parse_version


def prepare_parser_from_dict(dic, parser=None):
Expand Down Expand Up @@ -132,8 +134,13 @@ def parse_args_as_dict(parser, return_plain_args=False, args=None):
for group in parser._action_groups:
group_dict = {a.dest: getattr(args, a.dest, None) for a in group._group_actions}
args_dic[group.title] = group_dict
args_dic["main_args"] = args_dic["optional arguments"]
del args_dic["optional arguments"]
opt_args_key = (
"optional arguments"
if parse_version(platform.python_version()) < parse_version("3.10")
else "options"
)
args_dic["main_args"] = args_dic[opt_args_key]
del args_dic[opt_args_key]
if return_plain_args:
return args_dic, args
return args_dic

0 comments on commit bd3caa3

Please sign in to comment.