install k8s in pytest gpu

Author: nitins17

.github/workflows/pytest_gpu.yml

@@ -43,6 +43,6 @@ jobs:
       - name: Install dependencies
         env:
           JAXCI_PYTHON: python${{ matrix.python }}
-        run: $JAXCI_PYTHON -m pip install --upgrade     numpy=="2.0.0" scipy=="1.13.1"     wheel build pytest-xdist absl-py opt-einsum colorama portpicker     matplotlib 'importlib_metadata>=4.6' hypothesis flatbuffers filelock
+        run: $JAXCI_PYTHON -m pip install --upgrade     numpy=="2.0.0" scipy=="1.13.1"     wheel build pytest-xdist absl-py opt-einsum colorama portpicker     matplotlib 'importlib_metadata>=4.6' hypothesis flatbuffers filelock k8s
       - name: Run Pytest GPU tests
         run: ./ci/run_pytest.sh "ci/envs/run_tests/pytest_gpu"