e10b29b425ec9f51ebe1dd30b01eaeea9d3a9c84,src/sagemaker/cli/compatibility/v2/modifiers/framework_version.py,FrameworkVersionEnforcer,modify_node,#FrameworkVersionEnforcer#Any#,81

Before Change

        // if framework_version is not supplied, get default and append keyword
        framework_version = _arg_value(node, FRAMEWORK_ARG)
        if framework_version is None:
            framework_version = FRAMEWORK_DEFAULTS[framework]
            node.keywords.append(ast.keyword(arg=FRAMEWORK_ARG, value=ast.Str(s=framework_version)))

        // if py_version is not supplied, get a conditional default, and if not None, append keyword
        py_version = _arg_value(node, PY_ARG)
        if py_version is None:
            py_version = _py_version_defaults(framework, framework_version, is_model)
            if py_version:

After Change

        framework, is_model = _framework_from_node(node)

        // if framework_version is not supplied, get default and append keyword
        if matching.has_arg(node, FRAMEWORK_ARG):
            framework_version = parsing.arg_value(node, FRAMEWORK_ARG)
        else:
            framework_version = FRAMEWORK_DEFAULTS[framework]
            node.keywords.append(ast.keyword(arg=FRAMEWORK_ARG, value=ast.Str(s=framework_version)))

        // if py_version is not supplied, get a conditional default, and if not None, append keyword
        if not matching.has_arg(node, PY_ARG):
            py_version = _py_version_defaults(framework, framework_version, is_model)
            if py_version:
                node.keywords.append(ast.keyword(arg=PY_ARG, value=ast.Str(s=py_version)))

Instances