From 0fcfeb4792721c6053c2fb59252c13eced0ab621 Mon Sep 17 00:00:00 2001
From: Zuby Afzal <git.zuby@gmail.com>
Date: Mon, 7 Jul 2025 17:16:03 -0700
Subject: [PATCH] Format all Java files with google-java-format; add
 directories to lint workflow

---
 .github/workflows/lint.yml                    |  7 ++++-
 .../RuntimeInstrumentationTest.java           | 26 +++++++++----------
 .../executorch/extension/llm/LlmModule.java   |  1 -
 3 files changed, 19 insertions(+), 15 deletions(-)

diff --git a/.github/workflows/lint.yml b/.github/workflows/lint.yml
index f4f0c91c4ad..fc896c5e970 100644
--- a/.github/workflows/lint.yml
+++ b/.github/workflows/lint.yml
@@ -83,8 +83,13 @@ jobs:
       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 \
           examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/*.java \
-          extension/benchmark/android/benchmark/app/src/main/java/org/pytorch/minibench/*.java)
+          examples/demo-apps/android/LlamaDemo/app/src/androidTest/java/com/example/executorchllamademo/*.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/"
diff --git a/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/RuntimeInstrumentationTest.java b/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/RuntimeInstrumentationTest.java
index 27114b4cc77..e65507a424b 100644
--- a/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/RuntimeInstrumentationTest.java
+++ b/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/RuntimeInstrumentationTest.java
@@ -11,27 +11,27 @@
 import static org.junit.Assert.assertNotNull;
 
 import androidx.test.ext.junit.runners.AndroidJUnit4;
-import org.junit.runner.RunWith;
 import org.junit.Test;
+import org.junit.runner.RunWith;
 
 /** Unit tests for {@link ExecuTorchRuntime}. */
 @RunWith(AndroidJUnit4.class)
 public class RuntimeInstrumentationTest {
 
-    @Test
-    public void testRuntimeApi() {
-        String[] ops = ExecuTorchRuntime.getRegisteredOps();
-        String[] backends = ExecuTorchRuntime.getRegisteredBackends();
+  @Test
+  public void testRuntimeApi() {
+    String[] ops = ExecuTorchRuntime.getRegisteredOps();
+    String[] backends = ExecuTorchRuntime.getRegisteredBackends();
 
-        assertNotNull(ops);
-        assertNotNull(backends);
+    assertNotNull(ops);
+    assertNotNull(backends);
 
-        for (String op : ops) {
-            assertNotNull(op);
-        }
+    for (String op : ops) {
+      assertNotNull(op);
+    }
 
-        for (String backend : backends) {
-            assertNotNull(backend);
-        }
+    for (String backend : backends) {
+      assertNotNull(backend);
     }
+  }
 }
diff --git a/extension/android/executorch_android/src/main/java/org/pytorch/executorch/extension/llm/LlmModule.java b/extension/android/executorch_android/src/main/java/org/pytorch/executorch/extension/llm/LlmModule.java
index 5c8a867514e..b014ceb75d8 100644
--- a/extension/android/executorch_android/src/main/java/org/pytorch/executorch/extension/llm/LlmModule.java
+++ b/extension/android/executorch_android/src/main/java/org/pytorch/executorch/extension/llm/LlmModule.java
@@ -11,7 +11,6 @@
 import com.facebook.jni.HybridData;
 import com.facebook.jni.annotations.DoNotStrip;
 import java.io.File;
-
 import org.pytorch.executorch.ExecuTorchRuntime;
 import org.pytorch.executorch.annotations.Experimental;