Added build args for codeflare-sdk in notebook image build (#167)

* Added build args for codeflare-sdk in notebook image build --------- Co-authored-by: Kevin Postlethwait <[email protected]>
project-codeflare · Jun 21, 2023 · ac02248 · ac02248
1 parent b4d84c1
commit ac02248
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom-nb-image/Dockerfile b/custom-nb-image/Dockerfile
@@ -13,7 +13,6 @@
 # limitations under the License.
 
 FROM  quay.io/opendatahub/notebooks:jupyter-minimal-ubi8-python-3.8-4c8f26e
-
 # Install: torch (v1.12), ray (v2.1.0) and others
 
 COPY requirements.txt requirements.txt
@@ -23,7 +22,8 @@ RUN pip install -r requirements.txt
 RUN pip uninstall pickle5 -y
 
 # Install codeflare-sdk and other libraries
-RUN pip install codeflare-sdk==0.4.5 \
+ARG SDK_VERSION=0.*
+RUN pip install codeflare-sdk==${SDK_VERSION} \
 				datasets==2.6.1 \
 				transformers==4.23.1 \
 				evaluate==0.3.0