Skip to content

Commit

Permalink
updated darts and sktime versions for base images
Browse files Browse the repository at this point in the history
  • Loading branch information
attila-balint-kul committed Feb 28, 2024
1 parent c3185bb commit c8fe71a
Show file tree
Hide file tree
Showing 7 changed files with 39 additions and 13 deletions.
40 changes: 33 additions & 7 deletions Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -93,42 +93,68 @@ MODEL_NAME := sf-naive
IMAGE_TAG := $(ENFOBENCH_VERSION)-$(MODEL_NAME)
DEFAULT_PORT := 3000

DARTS_VERSION := 0.27.0
SKTIME_VERSION := 0.24.1
DARTS_VERSION := 0.27.2
SKTIME_VERSION := 0.26.1
STATSFORECAST_VERSION := 1.5.0

.PHONY: base-images
## Create docker base image
base-images:

.PHONY: base-image-darts
## Build base image for darts
base-image-darts:
docker build --build-arg DARTS_VERSION=$(DARTS_VERSION) -t $(DOCKER_HUB_REPOSITORY):base-u8darts-$(DARTS_VERSION) ./docker/base/darts


.PHONY: base-image-sktime
## Build base image for sktime
base-image-sktime:
docker build --build-arg SKTIME_VERSION=$(SKTIME_VERSION) -t $(DOCKER_HUB_REPOSITORY):base-sktime-$(SKTIME_VERSION) ./docker/base/sktime


.PHONY: base-image-statsforecast
## Build base image for statsforecast
base-image-statsforecast:
docker build --build-arg STATSFORECAST_VERSION=$(STATSFORECAST_VERSION) -t $(DOCKER_HUB_REPOSITORY):base-statsforecast-$(STATSFORECAST_VERSION) ./docker/base/statsforecast


## Build base images
base-images: base-image-darts \
base-image-sktime \
base-image-statsforecast


.PHONY: push-base-images
## Push base images to Docker Hub
push-base-images:
docker push $(DOCKER_HUB_REPOSITORY):base-u8darts-$(DARTS_VERSION)
docker push $(DOCKER_HUB_REPOSITORY):base-sktime-$(SKTIME_VERSION)
docker push $(DOCKER_HUB_REPOSITORY):base-statsforecast-$(STATSFORECAST_VERSION)


.PHONY: image
## Create docker image
## Build a model image
image:
docker build -t $(DOCKER_HUB_REPOSITORY):$(IMAGE_TAG) ./models/$(MODEL_NAME)


.PHONY: push-image
## Push docker image to Docker Hub
## Push a model image to Docker Hub
push-image: image
docker push $(DOCKER_HUB_REPOSITORY):$(IMAGE_TAG)


.PHONY: run-image
## Run a model image
run-image: image
docker run -it --rm -p $(DEFAULT_PORT):3000 $(DOCKER_HUB_REPOSITORY):$(IMAGE_TAG)


MODELS = $(shell ls -d ./models/* | xargs -n 1 basename)
## Make all model images
images:
$(foreach var,$(MODELS), $(MAKE) image MODEL_NAME=$(var);)


## Push images to Docker Hub
push-images:
$(foreach var,$(MODELS), $(MAKE) push-image MODEL_NAME=$(var);)

Expand Down
2 changes: 1 addition & 1 deletion models/dt-daily-naive/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
FROM attilabalint/enfobench-demand-models:base-u8darts-0.27.0
FROM attilabalint/enfobench-demand-models:base-u8darts-0.27.2

WORKDIR /usr/local/app

Expand Down
2 changes: 1 addition & 1 deletion models/dt-four-theta/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
FROM attilabalint/enfobench-demand-models:base-u8darts-0.27.0
FROM attilabalint/enfobench-demand-models:base-u8darts-0.27.2

WORKDIR /usr/local/app

Expand Down
2 changes: 1 addition & 1 deletion models/dt-lightgbm/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
FROM attilabalint/enfobench-demand-models:base-u8darts-0.27.0
FROM attilabalint/enfobench-demand-models:base-u8darts-0.27.2

# OpenMP is required by LightGBM
RUN apt-get update && \
Expand Down
2 changes: 1 addition & 1 deletion models/dt-linear-regression/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
FROM attilabalint/enfobench-demand-models:base-u8darts-0.27.0
FROM attilabalint/enfobench-demand-models:base-u8darts-0.27.2

WORKDIR /usr/local/app

Expand Down
2 changes: 1 addition & 1 deletion models/dt-naive-mean/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
FROM attilabalint/enfobench-demand-models:base-u8darts-0.27.0
FROM attilabalint/enfobench-demand-models:base-u8darts-0.27.2

WORKDIR /usr/local/app

Expand Down
2 changes: 1 addition & 1 deletion models/dt-theta/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
FROM attilabalint/enfobench-demand-models:base-u8darts-0.27.0
FROM attilabalint/enfobench-demand-models:base-u8darts-0.27.2

WORKDIR /usr/local/app

Expand Down

0 comments on commit c8fe71a

Please sign in to comment.