add k-means-constrained==0.7.3 to image

richmanbtc · Jul 19, 2023 · 556825c · 556825c
1 parent 5166732
commit 556825c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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 \