[easy] Fix upload test stats after master -> main switch (#99924)

Fixes #ISSUE_NUMBER

Pull Request resolved: https://github.com/pytorch/pytorch/pull/99924
Approved by: https://github.com/huydhn
diff --git a/tools/stats/upload_test_stats.py b/tools/stats/upload_test_stats.py
index d6eaf40..310ecd2 100644
--- a/tools/stats/upload_test_stats.py
+++ b/tools/stats/upload_test_stats.py
@@ -369,8 +369,8 @@
         failed_tests_cases,
     )
 
-    if args.head_branch == "master":
-        # For master jobs, upload everything.
+    if args.head_branch == "main":
+        # For jobs on main branch, upload everything.
         upload_workflow_stats_to_s3(
             args.workflow_run_id, args.workflow_run_attempt, "test_run", test_cases
         )