From 0fcb376281f2e133465652628f7a7ede031a0ce6 Mon Sep 17 00:00:00 2001
From: Abinila Siva <163017635+abinila4@users.noreply.github.com>
Date: Thu, 22 May 2025 16:07:38 -0400
Subject: [PATCH] Reduce Docker size: remove extra MemryX deps

---
 docker/main/Dockerfile | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/docker/main/Dockerfile b/docker/main/Dockerfile
index 3bfcb9948..591070ae6 100644
--- a/docker/main/Dockerfile
+++ b/docker/main/Dockerfile
@@ -177,7 +177,13 @@ RUN wget -q https://bootstrap.pypa.io/get-pip.py -O get-pip.py \
     && python3 get-pip.py "pip"
 
 # Build MemryX SDK wheel
-RUN pip3 wheel --wheel-dir=/wheels/memx/ --extra-index-url https://developer.memryx.com/pip memryx~=1.2.0
+RUN pip3 wheel --wheel-dir=/wheels/memx/ \
+     --index-url https://download.pytorch.org/whl/cpu \
+     --extra-index-url https://developer.memryx.com/pip \
+     --extra-index-url https://pypi.org/simple \
+    memryx~=1.2.0
+
+RUN rm -f /wheels/memx/PySide6* /wheels/memx/Qt* /wheels/memx/qt_material*
 
 COPY docker/main/requirements.txt /requirements.txt
 RUN pip3 install -r /requirements.txt