From 4da199f83d98a9372fb294bf49d5ac16945ed58c Mon Sep 17 00:00:00 2001
From: Michael Hudgins <michaelhudgins@google.com>
Date: Mon, 23 Sep 2024 17:33:07 +0000
Subject: [PATCH] Move ml-dtypes upgrade further down

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

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 166f188c..6b9f17a2 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -31,13 +31,13 @@ jobs:
     -   name: Setup Released JAX
         run: |
           pip install "jax[cuda12]"
-          # Need newer ml-dtypes because we install newer numpy
-          pip install --upgrade ml-dtypes
           pip install pytest
     -   name: Test JAX Triton
         run: |
           echo "Running JAX Triton GPU Tests"
           nvidia-smi
           pip install . 
+          # Need newer ml-dtypes because we install newer numpy
+          pip install --upgrade ml-dtypes
           pytest -v --tb=short tests/