5461fe7792ce43c3a7009a602c086842fe9fae64,tools/nni_annotation/search_space_generator.py,SearchSpaceGenerator,visit_Call,#SearchSpaceGenerator#Any#,51

Before Change

        if func in ("choice", "function_choice"):
            // arguments of `choice` may contain complex expression,
            // so use indices instead of arguments
            args = list(range(len(node.args)))
        else:
            // arguments of other functions must be literal number
            assert all(type(arg) is ast.Num for arg in node.args), "Smart parameter\"s arguments must be number literals"
            args = [arg.n for arg in node.args]

After Change

            // we will use keys in the dict as the choices, which is generated by code_generator according to the args given by user
            assert len(node.args) == 1, "Smart parameter has arguments other than dict"
            // check if it is a number or a string and get its value accordingly
            args = [key.n if type(key) is ast.Num else key.s for key in node.args[0].keys]
        else:
            // arguments of other functions must be literal number
            assert all(type(arg) is ast.Num for arg in node.args), "Smart parameter\"s arguments must be number literals"

Instances