chore: import upstream snapshot with attribution

release/util/sanity_check.py

@@ -0,0 +1,40 @@
+import sys
+
+import click
+
+import ray
+
+
+@click.command()
+@click.option("--ray_version", required=True, type=str)
+@click.option("--ray_commit", required=True, type=str)
+def main(ray_version, ray_commit):
+    print("Sanity check python version: {}".format(sys.version))
+    assert (
+        ray_version == ray.__version__
+    ), "Given Ray version {} is not matching with downloaded version {}".format(
+        ray_version, ray.__version__
+    )
+    assert (
+        ray_commit == ray.__commit__
+    ), "Given Ray commit {} is not matching with downloaded version {}".format(
+        ray_commit, ray.__commit__
+    )
+    assert ray.__file__ is not None
+
+    ray.init()
+    assert ray.is_initialized()
+
+    @ray.remote
+    def return_arg(arg):
+        return arg
+
+    val = 3
+    print("Running basic sanity check.")
+    assert ray.get(return_arg.remote(val)) == val
+    ray.shutdown()
+    print("Sanity check succeeded on Python {}".format(sys.version))
+
+
+if __name__ == "__main__":
+    main()

release/util/sanity_check_cpp.sh

@@ -0,0 +1,13 @@
+#!/usr/bin/env bash
+
+# This script generate a ray C++ template and run example
+set -e
+rm -rf ray-template
+ray cpp --generate-bazel-project-template-to ray-template
+(
+    cd ray-template
+
+    # Our generated CPP template does not work with bazel 7.x ,
+    # so pin the bazel version to 6
+    USE_BAZEL_VERSION=6.5.0 bash run.sh
+)

release/util/sanity_check_windows.sh

@@ -0,0 +1,67 @@
+#!/usr/bin/env bash
+
+# This script automatically download ray and run the sanity check for Windows
+# in various Python version. This script requires conda command to exist.
+
+set -euo pipefail
+
+export RAY_HASH="${RAY_HASH:-}"
+export RAY_VERSION="${RAY_VERSION:-}"
+
+if [[ -z "$RAY_HASH" ]]; then
+    echo "RAY_HASH env var should be provided"
+    exit 1
+fi
+
+if [[ -z "$RAY_VERSION" ]]; then
+    echo "RAY_VERSION env var should be provided"
+    exit 1
+fi
+
+if [[ ! -x "$(command -v conda)" ]]; then
+    echo "conda doesn't exist. Please download conda for this machine"
+    exit 1
+else
+    echo "conda exists"
+fi
+
+echo "Start downloading Ray version ${RAY_VERSION} of commit ${RAY_HASH}"
+
+python -m pip install --upgrade pip
+
+# This is required to use conda activate
+source "$(conda info --base)/etc/profile.d/conda.sh"
+
+PYTHON_VERSIONS=( "3.10" "3.11" "3.12" )
+
+for PYTHON_VERSION in "${PYTHON_VERSIONS[@]}"; do
+    ENV_NAME="${RAY_VERSION}-${PYTHON_VERSION}-env"
+    conda create -y -n "${ENV_NAME}" python="${PYTHON_VERSION}"
+    conda activate "${ENV_NAME}"
+
+    FAILED="false"
+
+    GOT_PYTHON_VERSION="$(python -c 'import sys; v=sys.version_info; print(f"{v.major}.{v.minor}")')"
+    if [[ "${GOT_PYTHON_VERSION}" != "${PYTHON_VERSION}" ]]; then
+        echo "Python version incorrect, got ${GOT_PYTHON_VERSION}, want ${PYTHON_VERSION}"
+        FAILED="true"
+    else
+        pip install --index-url 'https://test.pypi.org/simple/' --extra-index-url 'https://pypi.org/simple' "ray[cpp]==${RAY_VERSION}"
+
+        printf "\n\n\n"
+        echo "========================================================="
+        python sanity_check.py --ray_version="$RAY_VERSION" --ray_commit="$RAY_HASH"
+        echo "========================================================="
+        printf "\n\n\n"
+    fi
+
+    conda deactivate
+    conda remove -y --name "${ENV_NAME}" --all
+
+    if [[ "$FAILED" == "true" ]]; then
+        echo "PYTHON ${PYTHON_VERSION} failed sanity check."
+        exit 1
+    else
+        echo "PYTHON ${PYTHON_VERSION} succeed sanity check."
+    fi
+done