Replace XGBoost image for E2E with community hosted

Signed-off-by: Yuki Iwai <yuki.iwai.tz@gmail.com>

.github/workflows/integration-tests.yaml

@@ -55,6 +55,22 @@ jobs:
             python-version: "3.10"
 
     steps:
+      # This step is a Workaround to avoid the "No space left on device" error.
+      # ref: https://github.com/actions/runner-images/issues/2840
+      - name: Remove unnecessary files
+        shell: bash
+        run: |
+          sudo rm -rf /usr/share/dotnet
+          sudo rm -rf /opt/ghc
+          sudo rm -rf "/usr/local/share/boost"
+          sudo rm -rf "$AGENT_TOOLSDIRECTORY"
+          sudo rm -rf /usr/local/lib/android
+          sudo rm -rf /usr/local/share/powershell
+          sudo rm -rf /usr/share/swift
+          
+          echo "Disk usage after cleanup:"
+          df -h
+
       - name: Checkout
         uses: actions/checkout@v3
 
@@ -90,6 +106,11 @@ jobs:
           GANG_SCHEDULER_NAME: ${{ matrix.gang-scheduler-name }}
           KUBERNETES_VERSION: ${{ matrix.kubernetes-version }}
 
+#      - name: Pre-load job images
+#        run: |
+#          docker pull docker.io/kubeflow/xgboost-dist-iris:latest
+#          kind load docker-image docker.io/kubeflow/xgboost-dist-iris:latest --name training-operator-cluster
+
       - name: Run tests
         run: |
           pip install pytest

sdk/python/test/e2e/test_e2e_xgboostjob.py

@@ -181,7 +181,8 @@ def generate_xgboostjob(
 def generate_container() -> V1Container:
     return V1Container(
         name=CONTAINER_NAME,
-        image="docker.io/merlintang/xgboost-dist-iris:1.1",
+        image="docker.io/kubeflow/xgboost-dist-iris:latest",
+        # image_pull_policy="IfNotPresent",
         args=[
             "--job_type=Train",
             "--xgboost_parameter=objective:multi:softprob,num_class:3",