Add torchvision as dependency for image (#1131)

* Enhance RayClient shutdown process by resetting RAY_ADDRESS environment variable and improve logging for Grafana and Prometheus dashboard access with dynamic port information.

Signed-off-by: Abhinav Garg <[email protected]>

* move ray address around

Signed-off-by: Praateek <[email protected]>

* accept ip address as arg

Signed-off-by: Praateek <[email protected]>

* add torchvision

Signed-off-by: Praateek <[email protected]>

---------

Signed-off-by: Abhinav Garg <[email protected]>
Signed-off-by: Praateek <[email protected]>
Co-authored-by: Abhinav Garg <[email protected]>

Author: praateekmahajan

pyproject.toml

@@ -75,10 +75,13 @@ audio_cuda12 = [
     "nemo_curator[cuda12]",
 ]
 
-image_cpu = []
+image_cpu = [
+    "torchvision"
+]
 
 # NVIDIA DALI (simplified; update the package to match your CUDA version if needed)
 image_cuda12 = [
+    "nemo_curator[image_cpu]",
     "nemo_curator[cuda12]",
     "nemo_curator[deduplication_cuda12]",
     "nvidia-dali-cuda120",