diff --git a/ci/jenkins_tests/run_tune_tests.sh b/ci/jenkins_tests/run_tune_tests.sh index 8b8fe0160..1cc12c0cc 100755 --- a/ci/jenkins_tests/run_tune_tests.sh +++ b/ci/jenkins_tests/run_tune_tests.sh @@ -97,8 +97,8 @@ $SUPPRESS_OUTPUT docker run --rm --shm-size=${SHM_SIZE} --memory=${MEMORY_SIZE} python /ray/python/ray/tune/examples/hyperopt_example.py \ --smoke-test -# $SUPPRESS_OUTPUT docker run --rm --shm-size=${SHM_SIZE} --memory=${MEMORY_SIZE} --memory-swap=-1 $DOCKER_SHA \ -# python /ray/doc/source/tune/_tutorials/tune-sklearn.py +$SUPPRESS_OUTPUT docker run --rm --shm-size=${SHM_SIZE} --memory=${MEMORY_SIZE} --memory-swap=-1 $DOCKER_SHA \ + python /ray/doc/source/tune/_tutorials/tune-sklearn.py # if [[ ! -z "$SIGOPT_KEY" ]]; then # $SUPPRESS_OUTPUT docker run --rm --shm-size=${SHM_SIZE} --memory=${MEMORY_SIZE} --memory-swap=-1 -e SIGOPT_KEY $DOCKER_SHA \ diff --git a/doc/requirements-doc.txt b/doc/requirements-doc.txt index dcddb9606..6cec0b448 100644 --- a/doc/requirements-doc.txt +++ b/doc/requirements-doc.txt @@ -24,4 +24,5 @@ sphinx_rtd_theme tabulate uvicorn werkzeug -git+https://github.com/ray-project/tune-sklearn.git#egg=tune-sklearn +tune-sklearn==0.0.5 +scikit-optimize diff --git a/docker/tune_test/requirements.txt b/docker/tune_test/requirements.txt index a797e0143..58294f5ce 100644 --- a/docker/tune_test/requirements.txt +++ b/docker/tune_test/requirements.txt @@ -32,4 +32,4 @@ xgboost zoopt>=0.4.0 timm dataclasses -git+https://github.com/ray-project/tune-sklearn.git#egg=tune-sklearn +tune-sklearn==0.0.5