MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / __init__

Method __init__

tools/Polygraphy/tests/tools/args/helper.py:23–38  ·  view source on GitHub ↗
(self, arg_group, deps=None)

Source from the content-addressed store, hash-verified

21
22class ArgGroupTestHelper:
23 def __init__(self, arg_group, deps=None):
24 self.deps = util.default(deps, [])
25
26 self.arg_group = arg_group
27 self.parser = argparse.ArgumentParser()
28
29 arg_groups = {type(self.arg_group): self.arg_group}
30 arg_groups.update({type(dep): dep for dep in self.deps})
31
32 for dep in self.deps:
33 dep.register(arg_groups)
34 self.arg_group.register(arg_groups)
35
36 for dep in self.deps:
37 dep.add_parser_args(self.parser)
38 self.arg_group.add_parser_args(self.parser)
39
40 def parse_args(self, cli_args):
41 args = self.parser.parse_args(cli_args)

Callers

nothing calls this directly

Calls 5

typeFunction · 0.85
defaultMethod · 0.45
updateMethod · 0.45
registerMethod · 0.45
add_parser_argsMethod · 0.45

Tested by

no test coverage detected