ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-13 07:08:24 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix fire binding

Browse Source
This commit is contained in:
Alex Burlacu 2023-03-23 13:20:28 +02:00
parent 90354a7751
commit 2604401dd3
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
clearml/binding/fire_bind.py
View File

@ -235,6 +235,9 @@ class PatchFire:
def __get_all_groups_and_commands(component, context):
groups = []
commands = {}
# skip modules
if inspect.ismodule(component):
return groups, commands
component_trace_result = PatchFire.__safe_Fire(component, [], PatchFire.__default_args, context)
group_args = [[]]
while len(group_args) > 0:
@ -256,7 +259,8 @@ class PatchFire:
@staticmethod
def __get_groups_and_commands_for_args(component, args_, parsed_flag_args, context, name=None):
component_trace = PatchFire.__safe_Fire(component, args_, parsed_flag_args, context, name=name)
groups, commands, _, _ = fire.helptext._GetActionsGroupedByKind(component_trace, verbose=False) # noqa
# set verbose to True or else we might miss some commands
groups, commands, _, _ = fire.helptext._GetActionsGroupedByKind(component_trace, verbose=True) # noqa
groups = [(name, member) for name, member in groups.GetItems()]
commands = [(name, member) for name, member in commands.GetItems()]
return groups, commands