diff --git a/images/base/Dockerfile b/images/base/Dockerfile
index d40c124..d384392 100644
--- a/images/base/Dockerfile
+++ b/images/base/Dockerfile
@@ -51,12 +51,14 @@ RUN pip install --no-cache-dir \
     web3==5.31.1 \
     catboost==1.1.1 \
     optuna==3.0.5 \
-    scipy==1.8.1
+    scipy==1.8.1 \
+    k-means-constrained==0.7.3
 
 # Version resolution takes too long when installed with other pips
 RUN pip install --no-cache-dir \
     mljar-supervised==0.11.5 \
-    autogluon==0.7.0
+    autogluon==0.7.0 \
+    tpot==0.11.7
 
 RUN git clone --recursive http://github.com/ibayer/fastFM.git /tmp/fastFM \
     && (cd /tmp/fastFM \