diff --git a/Dockerfile.hpu b/Dockerfile.hpu
index fe729c1..2fa8d3a 100644
--- a/Dockerfile.hpu
+++ b/Dockerfile.hpu
@@ -22,4 +22,5 @@ COPY requirements_hpu.txt /workspace/requirements_hpu.txt
 
 # Install Python packages
 RUN pip install --upgrade pip \
-    && pip install -r requirements_hpu.txt
+    && pip install -r requirements_hpu.txt \
+    && pip install -e .
\ No newline at end of file
diff --git a/requirements_hpu.txt b/requirements_hpu.txt
index 3eefe7c..5aba605 100644
--- a/requirements_hpu.txt
+++ b/requirements_hpu.txt
@@ -1,7 +1,3 @@
 -r requirements.txt
 optimum-habana==1.14.1
-transformers==4.45.2
-huggingface-hub==0.26.2
-tiktoken==0.8.0
-torch-geometric==2.6.1
-numba==0.60.0
+pytest