Skip to content

Commit 44dff2b

Browse files
GokuMohandaselliot-barn
authored andcommitted
removed adding the TaskPoolStrategy as it's not needed here (#57897)
Signed-off-by: elliot-barn <[email protected]>
1 parent 4e99cf0 commit 44dff2b

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

doc/source/ray-overview/examples/e2e-multimodal-ai-workloads/notebooks/01-Batch-Inference.ipynb

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -320,10 +320,7 @@
320320
"outputs": [],
321321
"source": [
322322
"# Add class.\n",
323-
"ds = ds.map(add_class,\n",
324-
" num_cpus=1,\n",
325-
" num_gpus=0,\n",
326-
" compute=ray.data.TaskPoolStrategy(size=4))\n"
323+
"ds = ds.map(add_class)\n"
327324
]
328325
},
329326
{

0 commit comments

Comments
 (0)