From e7d9cb6029a6cf9a7a3c45e8d894638d8181dd91 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Tue, 3 May 2022 16:39:52 -0500
Subject: [PATCH] Install jax and pygments for Gitlab CI

---
 .gitlab-ci.yml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 9dd03ad..11a78ad 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -2,7 +2,7 @@ Python 3 POCL:
   script:
   - export PY_EXE=python3
   - export PYOPENCL_TEST=portable:pthread
-  - export EXTRA_INSTALL="pyopencl mpi4py"
+  - export EXTRA_INSTALL="pyopencl mpi4py jax[cpu]"
   - curl -L -O https://gitlab.tiker.net/inducer/ci-support/raw/main/build-and-test-py-project.sh
   - ". ./build-and-test-py-project.sh"
   tags:
@@ -31,7 +31,7 @@ Python 3 POCL Examples:
 Pylint:
   script:
   - export PY_EXE=python3
-  - EXTRA_INSTALL="pyopencl mpi4py"
+  - EXTRA_INSTALL="pyopencl mpi4py jax[cpu] pygments"
   - curl -L -O https://gitlab.tiker.net/inducer/ci-support/raw/main/prepare-and-run-pylint.sh
   - . ./prepare-and-run-pylint.sh ${CI_PROJECT_NAME} test/test_*.py examples
   tags:
-- 
GitLab