Skip to content

Commit 8dc8816

Browse files
Alexandra BelousovAlexandra Belousov
Alexandra Belousov
authored and
Alexandra Belousov
committed
fix dask example
1 parent e7aac4c commit 8dc8816

File tree

2 files changed

+3
-8
lines changed

2 files changed

+3
-8
lines changed

examples/dask-preprocessing-and-training/dask_on_ray.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,9 @@ def read_taxi_df_dask(dataset_path, X_vars, y_vars):
3535

3636
img = rh.Image("dask_img").pip_install(
3737
[
38+
"dask[distributed]<2024.11.0",
39+
"dask[dataframe]<2024.11.0",
3840
"dask-ml",
39-
"dask[distributed]",
40-
"dask[dataframe]",
4141
"boto3",
4242
"s3fs",
4343
"xgboost",

examples/dask-preprocessing-and-training/launch_lightgbm_dask_aws.py

+1-6
Original file line numberDiff line numberDiff line change
@@ -6,12 +6,7 @@
66
cluster_name = f"rh-new-{num_nodes}"
77

88
img = rh.Image("dask-img").pip_install(
9-
[
10-
"dask[distributed,dataframe]",
11-
"dask-ml",
12-
"gcsfs",
13-
"lightgbm",
14-
],
9+
["dask[distributed,dataframe]", "dask-ml", "gcsfs", "lightgbm", "s3fs"],
1510
)
1611

1712
cluster = rh.compute(

0 commit comments

Comments
 (0)