Build fix hopefully

Kaggle · Dec 1, 2023 · 4c5f7cd · 4c5f7cd
1 parent 5a1504a
commit 4c5f7cd
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
@@ -364,6 +364,8 @@ RUN pip install mpld3 \
         eli5 \
         kaggle \
         kagglehub \
+        google-generativeai \
+        wrapt \
         mock \
         pytest && \
     /tmp/clean-layer.sh
@@ -635,10 +637,6 @@ RUN mkdir -p ~/src && git clone https://github.com/SohierDane/BigQuery_Helper ~/
     sed -i 's/)/packages=["bq_helper"])/g' ~/src/BigQuery_Helper/setup.py && \
     pip install -e ~/src/BigQuery_Helper && \
     /tmp/clean-layer.sh
-
-RUN pip install wrapt \
-        google-generativeai && \
-    /tmp/clean-layer.sh
 
 # Add BigQuery client proxy settings
 ENV PYTHONUSERBASE "/root/.local"