From d8e62cbba3547433243b148f414b37ed182e5772 Mon Sep 17 00:00:00 2001
From: Abhijeet Prasad <abhijeet@braintrustdata.com>
Date: Fri, 6 Mar 2026 12:15:48 -0500
Subject: [PATCH] ci: Test increasing shard counts

---
 .github/workflows/py.yaml | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/py.yaml b/.github/workflows/py.yaml
index 6e098fd1..ceff04e8 100644
--- a/.github/workflows/py.yaml
+++ b/.github/workflows/py.yaml
@@ -22,7 +22,7 @@ jobs:
       matrix:
         python-version: ["3.10", "3.11", "3.12", "3.13", "3.14"]
         os: [ubuntu-latest, windows-latest]
-        shard: [0, 1]
+        shard: [0, 1, 2, 3]
 
     steps:
       - uses: actions/checkout@v4
@@ -41,10 +41,10 @@ jobs:
       - name: Test whether the Python SDK can be imported
         run: |
           python -c 'import braintrust'
-      - name: Run nox tests (shard ${{ matrix.shard }}/2)
+      - name: Run nox tests (shard ${{ matrix.shard }}/4)
         shell: bash
         run: |
-          cd py && ./scripts/nox-matrix.sh ${{ matrix.shard }} 2
+          cd py && ./scripts/nox-matrix.sh ${{ matrix.shard }} 4
 
   adk-py:
     uses: ./.github/workflows/adk-py-test.yaml