vee1e/tflite-micro
1
0
Fork 0
mirror of https://github.com/vee1e/tflite-micro.git synced 2026-09-01 17:57:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Improve Check TfLite Files (#3566)

Browse source
This commit is contained in:
Esun Kim 2026-05-21 14:40:16 -07:00 • committed by GitHub
parent 9f5ac257ee
commit a9952453ab
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 33 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

8
.github/workflows/check_tflite_files.yml vendored
View file

@ -20,15 +20,17 @@ jobs:
check_tflite_files:
runs-on: ubuntu-latest
name: Check PR Modifies TfLite Files
permissions:
contents: read
pull-requests: read
steps:
- uses: actions/checkout@v6
with:
ref: ${{ inputs.trigger-sha }}
# Do NOT specify ref -> checks out target base branch (main) securely
- name: Check Files
if: ${{ !contains(inputs.pr-body, 'NO_CHECK_TFLITE_FILES=') }}
env:
GITHUB_REPOSITORY: ${{ github.repository }}
PR_NUMBER: ${{ inputs.pr-number }}
PR_SHA: ${{ inputs.trigger-sha }}
TFLM_BOT_TOKEN: ${{ secrets.tflm-bot-token || github.token }}
run: tensorflow/lite/micro/tools/ci_build/check_tflite_files.sh

2
.github/workflows/pr_test.yml vendored
View file

@ -64,6 +64,8 @@ jobs:
run: echo "CI Authorized."
call-check-tflite-files:
needs: [gatekeeper, approval-gate]
if: needs.gatekeeper.outputs.scope != 'none'
uses: ./.github/workflows/check_tflite_files.yml
with:
trigger-sha: ${{ github.event.pull_request.head.sha }}

33
tensorflow/lite/micro/tools/ci_build/check_tflite_files.sh
View file

@ -20,27 +20,46 @@
# Inputs:
# GITHUB_REPOSITORY
# PR_NUMBER
# PR_SHA (optional, to download tracking list from PR commit)
# TFLM_BOT_TOKEN
set -e
set -u
URL="https://api.github.com/repos/${GITHUB_REPOSITORY}/pulls/${PR_NUMBER}/files"
PR_FILES=$(curl -s -X GET -H "Authorization: Bearer ${TFLM_BOT_TOKEN}" "${URL}" | jq -r '.[] | .filename')
export GH_TOKEN="${TFLM_BOT_TOKEN}"
# Create a temp file for PR files
echo "Fetching files modified in PR #${PR_NUMBER}..."
# Use GitHub CLI auto-pagination to safely pull all file changes (up to 3,000)
PR_FILES=$(gh api "repos/${GITHUB_REPOSITORY}/pulls/${PR_NUMBER}/files" --paginate --jq '.[].filename')
# Create temp files for PR files list and PR's tflite_files database
TMP_PR_FILES=$(mktemp)
trap 'rm -f "${TMP_PR_FILES}"' EXIT
TMP_TFLITE_FILES=$(mktemp)
trap 'rm -f "${TMP_PR_FILES}" "${TMP_TFLITE_FILES}"' EXIT
echo "${PR_FILES}" > "${TMP_PR_FILES}"
if [ ! -f ci/tflite_files.txt ]; then
echo "Error: ci/tflite_files.txt not found!"
TFLITE_FILES_FILE="ci/tflite_files.txt"
if [ -n "${PR_SHA:-}" ]; then
echo "Downloading ci/tflite_files.txt from PR commit ${PR_SHA}..."
# Fetch via API (raw content accept header) to support secure reading of untrusted data
URL_TXT="repos/${GITHUB_REPOSITORY}/contents/ci/tflite_files.txt?ref=${PR_SHA}"
if gh api -H "Accept: application/vnd.github.v3.raw" "${URL_TXT}" > "${TMP_TFLITE_FILES}" 2>/dev/null; then
TFLITE_FILES_FILE="${TMP_TFLITE_FILES}"
echo "Successfully downloaded and using PR's version of tflite_files.txt."
else
echo "Warning: Could not download from PR commit. Falling back to base branch version."
fi
fi
if [ ! -f "${TFLITE_FILES_FILE}" ]; then
echo "Error: ${TFLITE_FILES_FILE} not found!"
exit 1
fi
# Check for intersection between PR files and TFLite files
CONFLICTS=$(grep -F -x -f ci/tflite_files.txt "${TMP_PR_FILES}" || true)
CONFLICTS=$(grep -F -x -f "${TFLITE_FILES_FILE}" "${TMP_PR_FILES}" || true)
if [ -n "${CONFLICTS}" ]; then
echo "The following files should be modified in the upstream Tensorflow repo:"