From 464dcb0d50f13f82bcc8a2879e37c1bc6a6bf8f5 Mon Sep 17 00:00:00 2001
From: shengguangming <shengguangming@bytedance.com>
Date: Sun, 1 Dec 2024 15:37:29 +0800
Subject: [PATCH] [ci] fix github action

---
 .github/workflows/yapf_format.yml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/.github/workflows/yapf_format.yml b/.github/workflows/yapf_format.yml
index f148839..8788acb 100644
--- a/.github/workflows/yapf_format.yml
+++ b/.github/workflows/yapf_format.yml
@@ -24,6 +24,8 @@ jobs:
         python-version: ["3.12"]
     steps:
       - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
+        with:
+          ref: ${{ github.head_ref }}  # Checkout the branch associated with the pull request
       - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@0b93645e9fea7318ecaed2b359559ac225c90a2b # v5.3.0
         with: