hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[AIR] Enable other notebooks previously marked with # REGRESSION (#26896)

Browse source 
Co-authored-by: Richard Liaw <rliaw@berkeley.edu>
This commit is contained in:
Jiao 2022-07-25 13:40:21 -07:00 • committed by GitHub
parent 5030a4c1d3
commit 5315f1e643
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
doc/source/ray-air/examples/BUILD
View file

@ -43,9 +43,7 @@ py_test_run_all_notebooks(
py_test_run_all_notebooks(
size = "large",
include = ["huggingface_text_classification.ipynb"],
exclude = [
"torch_incremental_learning.ipynb" # REGRESSION!
],
exclude = [],
data = ["//doc/source/ray-air/examples:air_examples"],
tags = ["exclusive", "team:ml", "gpu", "ray_air"],
)

2
doc/source/ray-air/examples/torch_incremental_learning.ipynb
View file

@ -1087,7 +1087,7 @@
"# Number of data parallel workers to use for training.\n",
"num_workers = 1\n",
"# Whether to use GPU or not.\n",
"use_gpu = False\n",
"use_gpu = ray.available_resources().get(\"GPU\", 0) > 0\n",
"\n",
"permuted_mnist = PermutedMNISTStream(n_tasks=n_tasks)\n",
"train_stream = permuted_mnist.generate_train_stream()\n",

4
doc/source/tune/examples/BUILD
View file

@ -19,9 +19,7 @@ py_test_run_all_notebooks(
exclude = [
"pbt_ppo_example.ipynb",
"tune-xgboost.ipynb",
"nyc_taxi_basic_processing.ipynb", # REGRESSION
"ocr_example.ipynb", # REGRESSION
"sigopt_example.ipynb", # REGRESSION
"sigopt_example.ipynb", # REGRESSION: no credentials
],
data = ["//doc/source/tune/examples:tune_examples"],
tags = ["exclusive", "team:ml"],