From cfa7e5408ba18162c9602f6e61445490e52d5e93 Mon Sep 17 00:00:00 2001
From: Catherine Lee <csl@fb.com>
Date: Fri, 13 Sep 2024 14:32:36 -0700
Subject: [PATCH] tc

---
 .github/workflows/update-test-times.yml | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/update-test-times.yml b/.github/workflows/update-test-times.yml
index ef59c6a6b9..00f2b396d4 100644
--- a/.github/workflows/update-test-times.yml
+++ b/.github/workflows/update-test-times.yml
@@ -23,12 +23,17 @@ jobs:
     steps:
       - uses: actions/checkout@v3
 
+      - name: Set up python 3.10
+        uses: actions/setup-python@v2
+        with:
+          python-version: 3.10
+
       - name: Install Dependencies
-        run: pip3 install boto3==1.19.12 clickhouse-connect==0.7.16
+        run: python -m install boto3==1.19.12 clickhouse-connect==0.7.16
 
       - name: Update test times
         run: |
-          python3 -m torchci.update_test_times
+          python -m torchci.update_test_times
         env:
           ROCKSET_API_KEY: ${{ secrets.ROCKSET_API_KEY }}
           CLICKHOUSE_ENDPOINT: ${{ secrets.CLICKHOUSE_HUD_USER_URL }}