feedback: GHA updates

Author: a9p

.github/workflows/simple-pbt-e2e-test.yaml

@@ -0,0 +1,33 @@
+name: E2E Test with simple-pbt
+on:
+  - pull_request
+
+jobs:
+  e2e:
+    runs-on: ubuntu-20.04
+    timeout-minutes: 120
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v2
+
+      - name: Setup Test Env
+        uses: ./.github/workflows/template-setup-e2e-test
+        with:
+          kubernetes-version: ${{ matrix.kubernetes-version }}
+
+      - name: Run e2e test with ${{ matrix.experiments }} experiments
+        uses: ./.github/workflows/template-e2e-test
+        with:
+          experiments: ${{ matrix.experiments }}
+          # Comma Delimited
+          trial-images: simple-pbt
+
+    strategy:
+      fail-fast: false
+      matrix:
+        # Detail: https://hub.docker.com/r/kindest/node
+        # TODO (tenzen-y): We need to consider running tests on more kubernetes versions.
+        # kubernetes-version: ["v1.20.15", "v1.21.12", "v1.22.9", "v1.23.6", "v1.24.1"]
+        kubernetes-version: ["v1.21.12", "v1.22.9", "v1.23.6"]
+        # Comma Delimited
+        experiments: ["simple-pbt"]

cmd/suggestion/pbt/v1beta1/Dockerfile

@@ -1,12 +1,17 @@
-FROM python:3.9
+FROM python:3.9-slim
 
 ENV TARGET_DIR /opt/katib
 ENV SUGGESTION_DIR cmd/suggestion/pbt/v1beta1
 ENV GRPC_HEALTH_PROBE_VERSION v0.4.6
 
 RUN if [ "$(uname -m)" = "ppc64le" ] || [ "$(uname -m)" = "aarch64" ]; then \
     apt-get -y update && \
-    apt-get -y install gfortran libopenblas-dev liblapack-dev && \
+    apt-get -y install gfortran libopenblas-dev liblapack-dev wget && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*; \
+    else \
+    apt-get -y update && \
+    apt-get -y install wget && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*; \
     fi

examples/v1beta1/hp-tuning/pbt.yaml renamed to examples/v1beta1/hp-tuning/simple-pbt.yaml

@@ -1,11 +1,11 @@
 apiVersion: kubeflow.org/v1beta1
 kind: Experiment
 metadata:
-  name: random-experiment
-  namespace: 'kubeflow'
+  namespace: kubeflow
+  name: simple-pbt
 spec:
-  maxTrialCount: 12
-  parallelTrialCount: 12
+  maxTrialCount: 2
+  parallelTrialCount: 2
   maxFailedTrialCount: 3
   resumePolicy: FromVolume
   objective:

examples/v1beta1/trial-images/simple-pbt/Dockerfile

@@ -1,4 +1,4 @@
-FROM python:3.9
+FROM python:3.9-slim
 
 ADD examples/v1beta1/trial-images/simple-pbt /opt/pbt
 WORKDIR /opt/pbt