From c0dcbce7099bb5ac26e0688d9a7e2ac75fb89005 Mon Sep 17 00:00:00 2001 From: Ran Ran Date: Tue, 12 Dec 2023 09:43:50 -0800 Subject: [PATCH] Disable JAX test runs (#1024) * Disable JAX test runs * Format the config --- tests/jax/compilation-cache.libsonnet | 3 +++ tests/jax/pod-test.libsonnet | 3 +++ 2 files changed, 6 insertions(+) diff --git a/tests/jax/compilation-cache.libsonnet b/tests/jax/compilation-cache.libsonnet index 7870b2442..5e4599100 100644 --- a/tests/jax/compilation-cache.libsonnet +++ b/tests/jax/compilation-cache.libsonnet @@ -20,6 +20,9 @@ local mixins = import 'templates/mixins.libsonnet'; compilationCacheTest:: common.JaxTest + common.tpuVmBaseImage + mixins.Functional { modelName: 'compilation-cache-test', + // Never trigger the run (Feb 31st does not exist) + schedule: '0 0 31 2 *', + setup: ||| pip install --upgrade pip diff --git a/tests/jax/pod-test.libsonnet b/tests/jax/pod-test.libsonnet index 255a27f35..02d457085 100644 --- a/tests/jax/pod-test.libsonnet +++ b/tests/jax/pod-test.libsonnet @@ -21,6 +21,9 @@ local tpus = import 'templates/tpus.libsonnet'; podTest:: common.JaxTest + mixins.Functional { modelName: 'pod-%s-%s' % [self.jaxlibVersion, self.tpuSettings.softwareVersion], + // Never trigger the run (Feb 31st does not exist) + schedule: '0 0 31 2 *', + setup: ||| %(installLocalJax)s %(maybeBuildJaxlib)s