mlflow_pip_deps.py

def mlflow_pip_deps() -> List[str]:
    try:
        poetry_export_result = subprocess.run(
            ["poetry", "export", "-f", "requirements.txt", "--without-hashes"],
            capture_output=True,
            encoding="utf8",
        ).stdout.splitlines()
        result = []
        for req in poetry_export_result:
            # ignore extra/index url lines
            if "index-url" in req:
                continue
            # remove the python version markers
            result.append(
                re.sub(
                    r'(( and )?python_version (>=|<=|==|<|>) "[\d.]+"( and)?)', "", req
                )
            )
        print(result)
        return result
    except Exception as ex:
        raise Exception(
            "Make sure the 'poetry-plugin-export' is installed. Error: {0}".format(ex)
        )