Skip to content

Commit b9a6c0e

Browse files
new parent image (#44)
* new parent image * adds xgboost, and lightgbm
1 parent 45c6c08 commit b9a6c0e

File tree

4 files changed

+15
-5
lines changed

4 files changed

+15
-5
lines changed

Dockerfile

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
1-
FROM kernai/refinery-parent-images:v1.12.2-torch-cpu
1+
FROM kernai/refinery-parent-images:v1.13.0-torch-cpu
22

33
RUN apt-get update && \
4-
apt-get install --no-install-recommends -y curl && \
4+
apt-get install --no-install-recommends -y curl libgomp1 && \
55
rm -rf /var/lib/apt/lists/*
66

77
COPY requirements.txt .

requirements.txt

+9-1
Original file line numberDiff line numberDiff line change
@@ -68,17 +68,21 @@ joblib==1.3.2
6868
# -r requirements/torch-cpu-requirements.txt
6969
# scikit-learn
7070
# scikit-optimize
71+
lightgbm==4.1.0
72+
# via -r requirements/requirements.in
7173
minio==7.1.12
7274
# via -r requirements/torch-cpu-requirements.txt
7375
numpy==1.23.4
7476
# via
7577
# -r requirements/torch-cpu-requirements.txt
78+
# lightgbm
7679
# pandas
7780
# scikit-learn
7881
# scikit-optimize
7982
# scipy
8083
# sequencelearn
8184
# transformers
85+
# xgboost
8286
packaging==23.1
8387
# via
8488
# -r requirements/torch-cpu-requirements.txt
@@ -138,9 +142,11 @@ scikit-optimize==0.9.0
138142
scipy==1.11.2
139143
# via
140144
# -r requirements/torch-cpu-requirements.txt
145+
# lightgbm
141146
# scikit-learn
142147
# scikit-optimize
143148
# sequencelearn
149+
# xgboost
144150
sequencelearn==0.0.9
145151
# via -r requirements/requirements.in
146152
six==1.16.0
@@ -183,11 +189,13 @@ typing-extensions==4.8.0
183189
# pydantic
184190
# starlette
185191
# torch
186-
urllib3==1.26.16
192+
urllib3==1.26.18
187193
# via
188194
# -r requirements/torch-cpu-requirements.txt
189195
# botocore
190196
# minio
191197
# requests
192198
uvicorn==0.22.0
193199
# via -r requirements/torch-cpu-requirements.txt
200+
xgboost==2.0.1
201+
# via -r requirements/requirements.in

requirements/requirements.in

+3-1
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,6 @@
22
#included in base requirements since 1.8.2 because of scikit-optimize
33
# scikit-learn==1.2.1
44
# scipy==1.9.0
5-
sequencelearn==0.0.9
5+
sequencelearn==0.0.9
6+
xgboost==2.0.1
7+
lightgbm==4.1.0

requirements/torch-cpu-requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ typing-extensions==4.8.0
130130
# pydantic
131131
# starlette
132132
# torch
133-
urllib3==1.26.16
133+
urllib3==1.26.18
134134
# via
135135
# botocore
136136
# minio

0 commit comments

Comments
 (0)