Function run_string_as_driver

python/ray/_common/test_utils.py:374–403

Run a driver as a separate process. Args: driver_script: A string to run as a Python script. env: The environment variables for the driver. encode: The encoding to use for the driver script. Returns: The script's output.

(
    driver_script: str, env: Dict = None, encode: str = "utf-8"
)

def run_string_as_driver(
    driver_script: str, env: Dict = None, encode: str = "utf-8"
) -> str:
    """Run a driver as a separate process.

    Args:
        driver_script: A string to run as a Python script.
        env: The environment variables for the driver.
        encode: The encoding to use for the driver script.

    Returns:
        The script's output.
    """
    proc = subprocess.Popen(
        [sys.executable, "-"],
        stdin=subprocess.PIPE,
        stdout=subprocess.PIPE,
        stderr=subprocess.STDOUT,
        env=env,
    )
    with proc:
        output = proc.communicate(driver_script.encode(encoding=encode))[0]
        if proc.returncode:
            print(decode(output, encode_type=encode))
            logger.error(proc.stderr)
            raise subprocess.CalledProcessError(
                proc.returncode, proc.args, output, proc.stderr
            )
        out = decode(output, encode_type=encode)
    return out


@dataclass