c08e45b9968501980c5e5daab2a46c3b7305c162,setup.py,,update_tf_dep,#Any#,441

Before Change

            return
        if custom_tf:
            REQUIRED_PACKAGES.append(custom_tf)
        return

    tf_ver = "tensorflow-gpu=={}.0".format(tf_ver)
    REQUIRED_PACKAGES.append(tf_ver)

After Change

    if not custom_tf:
        return

    custom_tf = os.path.realpath(os.path.expanduser(custom_tf))
    if not os.path.isfile(custom_tf):
        out_error("{} not found".format(custom_tf))
    elif os.path.splitext(custom_tf)[1] != ".whl":
        out_error("{} is not a valid pip wheel".format(custom_tf))

Instances