increase to 20 partitions

samos123 · Jun 27, 2024 · 30aec7e · 30aec7e
1 parent 4750f6a
commit 30aec7e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/set_matrix.py b/.github/workflows/set_matrix.py
@@ -1,13 +1,13 @@
 """
 A script to split up test files into partitions for GitHub Actions
 
-This allows us to run 10 partitions of tests at the same time on different
+This allows us to run 20 partitions of tests at the same time on different
 GitHub action runners.
 """
 import glob
 import json
 
-partitions = 10
+partitions = 20
 test_files = glob.glob("axlearn/**/*_test.py")
 # TODO(samos123): Figure out why clip is broken
 excluded_tests = ["axlearn/vision/clip_test.py"]