fix(ci): pin kubernetes<36 in github workflow

.github/workflows/e2e-test-train-api.yaml

@@ -53,7 +53,7 @@ jobs:
 
       - name: Run tests
         run: |
-          pip install pytest
+          pip install pytest "kubernetes<36"
           python3 -m pip install -e sdk/python[huggingface]
           pytest -s sdk/python/test/e2e-fine-tune-llm/test_e2e_pytorch_fine_tune_llm.py --log-cli-level=debug
         env:

.github/workflows/integration-tests.yaml

@@ -80,7 +80,7 @@ jobs:
 
       - name: Run tests
         run: |
-          pip install pytest
+          pip install pytest "kubernetes<36"
           python3 -m pip install -e sdk/python; pytest -s sdk/python/test/e2e --log-cli-level=debug --namespace=default
         env:
           GANG_SCHEDULER_NAME: ${{ matrix.gang-scheduler-name }}

.github/workflows/test-example-notebooks.yaml

@@ -28,7 +28,7 @@ jobs:
 
       - name: Install Python Dependencies
         run: |
-          pip install papermill==2.6.0 jupyter==1.1.1 ipykernel==6.29.5
+          pip install papermill==2.6.0 jupyter==1.1.1 ipykernel==6.29.5 "kubernetes<36" "fastjsonschema<2.22"
 
       - name: Run Jupyter Notebook with Papermill
         shell: bash

.github/workflows/test-python.yaml

@@ -28,7 +28,12 @@ jobs:
 
       - name: Install dependencies
         run: |
-          pip install pytest python-dateutil urllib3 kubernetes
+          # TODO: kubernetes 36.0.0 switched to Pydantic models, breaking the
+          # FakeResponse deserialization hack in the SDK unit tests. To remove
+          # this pin, either adapt FakeResponse in
+          # sdk/python/kubeflow/training/utils/utils.py for Pydantic or mock
+          # ApiClient.deserialize directly in the test suite.
+          pip install pytest python-dateutil urllib3 "kubernetes<36"
           pip install -U './sdk/python[huggingface]'
 
       - name: Run unit test for training sdk

sdk/python/test/e2e/test_e2e_jaxjob.py

@@ -156,5 +156,5 @@ def generate_container() -> V1Container:
     return V1Container(
         name=CONTAINER_NAME,
         image=os.getenv("JAX_JOB_IMAGE", "docker.io/kubeflow/jaxjob-dist-spmd-mnist:latest"),
-        resources=V1ResourceRequirements(limits={"memory": "3Gi", "cpu": "1.2"}),
+        resources=V1ResourceRequirements(limits={"memory": "5Gi", "cpu": "1.2"}),
     )