Update the base image to easydl/dlrover:ci

intelligent-machine-learning · workingloong · Sep 22, 2023 · Sep 22, 2023 · Sep 22, 2023 · Sep 22, 2023
commit 8239dee8d22f2dba038cb6e761a22a85de4645f3
diff --git a/.github/actions/pre-commit/action.yml b/.github/actions/pre-commit/action.yml
@@ -3,7 +3,7 @@ name: pre-commit
 description: run pre-commit to check codes
 runs:
  using: 'docker'
- image: "easydl/easydl:ci"
+ image: "easydl/dlrover:ci"
  args:
  - "/bin/bash"
  - "-c"

diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -1,4 +1,4 @@
-FROM easydl/easydl:ci as builder
+FROM easydl/dlrover:ci as builder
 
 WORKDIR /dlrover
 COPY ./ .

diff --git a/docker/ci.dockerfile b/docker/ci.dockerfile
@@ -33,7 +33,8 @@ COPY docker/scripts/install-protobuf.bash /
 RUN /install-protobuf.bash && rm /install-protobuf.bash
 
 # Install Pre-commit
-RUN pip install pre-commit pytest -i https://mirrors.aliyun.com/pypi/simple/
+RUN pip install pre-commit pytest kubernetes grpcio-tools psutil \
+ deprecated -i https://mirrors.aliyun.com/pypi/simple/
 
 # Configure envtest for integration tests of kubebuilder
 ENV KUBEBUILDER_CONTROLPLANE_START_TIMEOUT 60s

diff --git a/examples/pytorch/mnist/mnist.dockerfile b/examples/pytorch/mnist/mnist.dockerfile
@@ -1,4 +1,4 @@
-FROM easydl/easydl:ci as builder
+FROM easydl/dlrover:ci as builder
 
 WORKDIR /dlrover
 COPY ./ .

diff --git a/examples/pytorch/nanogpt/nanogpt.dockerfile b/examples/pytorch/nanogpt/nanogpt.dockerfile
@@ -1,4 +1,4 @@
-FROM easydl/easydl:ci as builder
+FROM easydl/dlrover:ci as builder
 
 WORKDIR /dlrover
 COPY ./ .