From d3fd2e69fd9f36c5120774e2dfa865f10d327803 Mon Sep 17 00:00:00 2001
From: eugr <eugr@spark2.home.eugr.net>
Date: Mon, 24 Nov 2025 15:47:20 -0800
Subject: [PATCH] Updated Dockerfile with additional deps

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

diff --git a/Dockerfile b/Dockerfile
index 841c93c..18da715 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -13,6 +13,7 @@ ENV VLLM_BASE_DIR=/workspace/vllm
 # Added 'git', 'wget', and 'python3-pip' as they are required for the script steps
 RUN apt-get update && apt-get install -y \
     curl \
+    vim \
     cmake \
     build-essential \
     ninja-build \
@@ -20,6 +21,7 @@ RUN apt-get update && apt-get install -y \
     python3-pip \
     git \
     wget \
+    gnuplot \
     && rm -rf /var/lib/apt/lists/*
 
 # Setup Workspace
@@ -47,7 +49,7 @@ ARG CACHEBUST_DEPS=1
 RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130
 
 # Install Helper libraries
-RUN pip install xgrammar triton
+RUN pip install xgrammar triton termplotlib
 
 # Install FlashInfer
 # Note: Using the same index URLs as provided in your script