diff --git a/doc/source/ray-overview/examples/e2e-multimodal-ai-workloads/notebooks/01-Batch-Inference.ipynb b/doc/source/ray-overview/examples/e2e-multimodal-ai-workloads/notebooks/01-Batch-Inference.ipynb
index d22c7aafea2e..0d3fda2d7aa3 100644
--- a/doc/source/ray-overview/examples/e2e-multimodal-ai-workloads/notebooks/01-Batch-Inference.ipynb
+++ b/doc/source/ray-overview/examples/e2e-multimodal-ai-workloads/notebooks/01-Batch-Inference.ipynb
@@ -320,10 +320,7 @@
    "outputs": [],
    "source": [
     "# Add class.\n",
-    "ds = ds.map(add_class,\n",
-    "    num_cpus=1,\n",
-    "    num_gpus=0,\n",
-    "    compute=ray.data.TaskPoolStrategy(size=4))\n"
+    "ds = ds.map(add_class)\n"
    ]
   },
   {