From 55852f627e29fe2d7ff2a2d6195a21353f747c3a Mon Sep 17 00:00:00 2001
From: Naveen S <naveens@arizona.edu>
Date: Fri, 6 Mar 2026 08:38:54 -0700
Subject: [PATCH] Fix JAVA_HOME for ARM64

---
 Dockerfile | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index c204ae8..e067175 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -12,7 +12,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     procps \
     && rm -rf /var/lib/apt/lists/*
 
-ENV JAVA_HOME=/usr/lib/jvm/java-17-openjdk-amd64
+# Dynamically set Java home for amd64/arm64
+RUN ln -s /usr/lib/jvm/java-17-openjdk-$(dpkg --print-architecture) /usr/lib/jvm/default-java
+ENV JAVA_HOME=/usr/lib/jvm/default-java
 ENV PATH=$JAVA_HOME/bin:$PATH
 
 # Install Spark