From ecbd5a17cc874fb58b231bb7efc2f07a53115d45 Mon Sep 17 00:00:00 2001
From: Angel Rey <angel@monadical.com>
Date: Tue, 15 Aug 2023 16:32:38 -0500
Subject: [PATCH] chore: fix ray in mac

---
 docker-compose.yaml            | 11 ++++++++---
 morpheus-worker/ray/Dockerfile |  3 ++-
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/docker-compose.yaml b/docker-compose.yaml
index a21dae0f..f44660de 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -122,8 +122,13 @@ services:
               capabilities: [ gpu ]
 
   worker-ray:
-    build: ./morpheus-worker/ray
-    command: ray start --head --dashboard-host 0.0.0.0 --block
+    build: 
+      context: ./morpheus-worker/ray
+      # if you are using a mac, uncomment this arg
+      # args:
+      #   RAY_IMAGE_TAG: latest-gpu-aarch64
+    image: morpheus-worker:latest
+    command: ray start --head --num-gpus 1 --dashboard-host 0.0.0.0 --block
     ports:
       - "8000:8000"
       - "8265:8265"
@@ -134,7 +139,7 @@ services:
         condition: service_started
 
   worker-ray-deployer:
-    build: ./morpheus-worker/ray
+    image: morpheus-worker:latest
     command: bash -c "sleep 5 && ray job submit -- serve deploy models.yaml"
     environment:
       - RAY_ADDRESS=http://worker-ray:8265
diff --git a/morpheus-worker/ray/Dockerfile b/morpheus-worker/ray/Dockerfile
index c15d91d2..5e198974 100644
--- a/morpheus-worker/ray/Dockerfile
+++ b/morpheus-worker/ray/Dockerfile
@@ -1,4 +1,5 @@
-FROM rayproject/ray:latest-gpu
+ARG RAY_IMAGE_TAG="latest-gpu"
+FROM rayproject/ray:$RAY_IMAGE_TAG
 
 # Configuration defaults
 ENV VENV_NAME ".venv-docker"