Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
FROM kernai/refinery-parent-images:v1.23.0-torch-cpu
FROM kernai/refinery-parent-images:v2.0.0-torch-cpu

RUN apt-get update && \
apt-get install --no-install-recommends -y curl libgomp1 && \
Expand Down
11 changes: 6 additions & 5 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ argon2-cffi-bindings==21.2.0
# argon2-cffi
boto3==1.39.6
# via -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
botocore==1.39.6
botocore==1.39.10
# via
# -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
# boto3
Expand All @@ -46,7 +46,7 @@ click==8.2.1
# via
# -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
# uvicorn
fastapi==0.115.2
fastapi==0.116.1
# via -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
filelock==3.18.0
# via
Expand Down Expand Up @@ -93,7 +93,7 @@ joblib==1.5.1
# -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
# scikit-learn
# scikit-optimize
lightgbm==4.1.0
lightgbm==4.6.0
# via -r requirements/requirements.in
markupsafe==3.0.2
# via
Expand Down Expand Up @@ -173,7 +173,7 @@ requests==2.32.4
# -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
# huggingface-hub
# transformers
s3transfer==0.13.0
s3transfer==0.13.1
# via
# -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
# boto3
Expand Down Expand Up @@ -208,7 +208,7 @@ sniffio==1.3.1
# anyio
sqlalchemy==1.4.42
# via -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
starlette==0.40.0
starlette==0.47.2
# via
# -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-ml-exec-env/requirements/torch-cpu-requirements.txt
# fastapi
Expand Down Expand Up @@ -244,6 +244,7 @@ typing-extensions==4.14.1
# minio
# pydantic
# pydantic-core
# starlette
# torch
urllib3==2.5.0
# via
Expand Down
2 changes: 1 addition & 1 deletion requirements/requirements.in
Original file line number Diff line number Diff line change
Expand Up @@ -4,4 +4,4 @@
# scipy==1.9.0
sequencelearn==0.0.9
xgboost==2.0.1
lightgbm==4.1.0
lightgbm==4.6.0
9 changes: 5 additions & 4 deletions requirements/torch-cpu-requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ argon2-cffi-bindings==21.2.0
# via argon2-cffi
boto3==1.39.6
# via -r common-requirements.in
botocore==1.39.6
botocore==1.39.10
# via
# boto3
# s3transfer
Expand All @@ -30,7 +30,7 @@ charset-normalizer==3.4.2
# via requests
click==8.2.1
# via uvicorn
fastapi==0.115.2
fastapi==0.116.1
# via -r mini-requirements.in
filelock==3.18.0
# via
Expand Down Expand Up @@ -117,7 +117,7 @@ requests==2.32.4
# -r mini-requirements.in
# huggingface-hub
# transformers
s3transfer==0.13.0
s3transfer==0.13.1
# via boto3
safetensors==0.5.3
# via transformers
Expand All @@ -137,7 +137,7 @@ sniffio==1.3.1
# via anyio
sqlalchemy==1.4.42
# via -r common-requirements.in
starlette==0.40.0
starlette==0.47.2
# via fastapi
sympy==1.14.0
# via torch
Expand All @@ -161,6 +161,7 @@ typing-extensions==4.14.1
# minio
# pydantic
# pydantic-core
# starlette
# torch
urllib3==2.5.0
# via
Expand Down