add pip upgrade to fix flax install (#1028)

GoogleCloudPlatform · Dec 18, 2023 · 5ecd8f3 · 5ecd8f3
1 parent c0dcbce
commit 5ecd8f3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/pax/nightly/common.libsonnet b/tests/pax/nightly/common.libsonnet
@@ -27,6 +27,9 @@ local mixins = import 'templates/mixins.libsonnet';
       # .bash_logout sometimes causes a spurious bad exit code, remove it.
       rm .bash_logout
 
+      # update pip version
+      pip install --upgrade pip
+
       # check for nightly build
       gsutil cp gs://pax-on-cloud-tpu-project/wheels/%(buildDate)s/paxml*.whl .
       gsutil cp gs://pax-on-cloud-tpu-project/wheels/%(buildDate)s/praxis*.whl .