f31b7ad689b1435e76744af4ff443607643a37fd,studio/experiment.py,,create_experiment,#Any#Any#Any#Any#Any#Any#Any#Any#,116

Before Change

           int(resources_needed.get("gpus")) > 0 and \
           (pkg._key == "tensorflow" or pkg._key == "tf-nightly"):
            packages.insert(0, pkg._key + "-gpu==" + pkg._version)
            packages.insert(0, pkg._key + "==" + pkg._version)

        elif pkg._key == "tensorflow-gpu" or pkg._key == "tf-nightly-gpu":
            packages.insert(0, pkg._key[:-4] + "==" + pkg._version)
        else:

After Change

        if pkg.startswith("-e git+"):
            // git package
            packages.append(pkg)
        elif "==" in pkg:
            // pypi package
            key = re.search(r"^.*?(?=\=\=)", pkg).group(0)
            version = re.search(r"(?<=\=\=).*\Z", pkg).group(0)

            if resources_needed is not None and \
                    int(resources_needed.get("gpus")) > 0:
                if (key == "tensorflow" or key == "tf-nightly"):
                    key = key + "-gpu"

            // TODO add installation logic for torch
            packages.append(key + "==" + version)

    return Experiment(
        key=key,
        filename=filename,

Instances