fix: Using version args to install the correct feast version (#3953)

* using version args to install the correct feast version Signed-off-by: Chester Ong <[email protected]> * revert the COPY command Signed-off-by: Chester Ong <[email protected]> --------- Signed-off-by: Chester Ong <[email protected]>
feast-dev · Feb 20, 2024 · b83a702 · b83a702
1 parent 6b8e96c
commit b83a702
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile b/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile
@@ -1,14 +1,17 @@
 FROM python:3.8
 
+# Input the feast version to install
+# This requires feast package to be available in pypi before building this image
+ARG VERSION
+
 RUN apt update && \
         apt install -y \
         jq \
         python3-dev \
         build-essential
 
 RUN pip install pip --upgrade
-RUN pip install "feast[aws,gcp,snowflake,redis,go,mysql,postgres]"
-
+RUN pip install "feast[aws,gcp,snowflake,redis,go,mysql,postgres]==${VERSION}"
 
 RUN apt update
 RUN apt install -y -V ca-certificates lsb-release wget

diff --git a/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile.dev b/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile.dev
@@ -1,5 +1,9 @@
 FROM python:3.8
 
+# Input the feast version to install
+# This requires feast package to be available in pypi before building this image
+ARG VERSION
+
 RUN apt update && \
         apt install -y \
         jq \
@@ -9,11 +13,11 @@ RUN apt update && \
 RUN pip install pip --upgrade
 COPY . .
 
-RUN pip install "feast[aws,gcp,snowflake,redis,go,mysql,postgres]"
+RUN pip install "feast[aws,gcp,snowflake,redis,go,mysql,postgres]==${VERSION}"
 
 RUN apt update
 RUN apt install -y -V ca-certificates lsb-release wget
 RUN wget https://apache.jfrog.io/artifactory/arrow/$(lsb_release --id --short | tr 'A-Z' 'a-z')/apache-arrow-apt-source-latest-$(lsb_release --codename --short).deb
 RUN apt install -y -V ./apache-arrow-apt-source-latest-$(lsb_release --codename --short).deb
 RUN apt update
-RUN apt -y install libarrow-dev
+RUN apt -y install libarrow-dev