Fix android lint

Author: mergennachin

.github/workflows/lint.yml

@@ -143,19 +143,24 @@ jobs:
       ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
       timeout: 90
       script: |
-        FILES_NEEDS_FORMAT=$(/opt/google-java-format -n \
-          extension/android/executorch_android/src/main/java/org/pytorch/executorch/*.java \
-          extension/android/executorch_android/src/main/java/org/pytorch/executorch/extension/llm/*.java \
-          extension/android/executorch_android/src/main/java/org/pytorch/executorch/annotations/*.java \
-          extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/*.java \
-          extension/benchmark/android/benchmark/app/src/main/java/org/pytorch/minibench/*.java \
-          extension/benchmark/android/benchmark/app/src/androidTest/java/org/pytorch/minibench/*.java)
-        if [ -n "$FILES_NEEDS_FORMAT" ]; then
-          echo "Warning: The following files need formatting. Please use google-java-format."
-          echo "Use a binary from https://github.com/google/google-java-format/releases/"
-          echo "For example:"
-          echo "wget https://github.com/google/google-java-format/releases/download/v1.23.0/google-java-format_linux-x86-64"
-          echo "chmod +x google-java-format_linux-x86-64"
-          echo "./google-java-format_linux-x86-64 -i $FILES_NEEDS_FORMAT"
-          exit 1
+        FILES_TO_CHECK=$(find extension/android/executorch_android/src/main/java/org/pytorch/executorch \
+                      extension/android/executorch_android/src/main/java/org/pytorch/executorch/extension/llm \
+                      extension/android/executorch_android/src/main/java/org/pytorch/executorch/annotations \
+                      extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch \
+                      extension/benchmark/android/benchmark/app/src/main/java/org/pytorch/minibench \
+                      extension/benchmark/android/benchmark/app/src/androidTest/java/org/pytorch/minibench \
+                      -type f \( -name "*.java" -o -name "*.kt" \) 2>/dev/null)
+
+        # Only run formatter if files were found
+        if [ -n "$FILES_TO_CHECK" ]; then
+            FILES_NEEDS_FORMAT=$(/opt/google-java-format -n $FILES_TO_CHECK)
+          if [ -n "$FILES_NEEDS_FORMAT" ]; then
+            echo "Warning: The following files need formatting. Please use google-java-format."
+            echo "Use a binary from https://github.com/google/google-java-format/releases/"
+            echo "For example:"
+            echo "wget https://github.com/google/google-java-format/releases/download/v1.23.0/google-java-format_linux-x86-64"
+            echo "chmod +x google-java-format_linux-x86-64"
+            echo "./google-java-format_linux-x86-64 -i $FILES_NEEDS_FORMAT"
+            exit 1
+          fi
         fi