Fix android instrumentation (#10335)

Author: kirklandsign

extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/LlmModuleInstrumentationTest.java

@@ -97,16 +97,7 @@ public void onResult(String result) {
 
             @Override
             public void onStats(String stats) {
-                float tps = 0;
-                try {
-                    JSONObject jsonObject = new JSONObject(stats);
-                    int numGeneratedTokens = jsonObject.getInt("generated_tokens");
-                    int inferenceEndMs = jsonObject.getInt("inference_end_ms");
-                    int promptEvalEndMs = jsonObject.getInt("prompt_eval_end_ms");
-                    tps = (float) numGeneratedTokens / (inferenceEndMs - promptEvalEndMs) * 1000;
-                    LlmModuleInstrumentationTest.this.onStats(tps);
-                } catch (JSONException e) {
-                }
+                LlmModuleInstrumentationTest.this.onStats(stats);
             }
         });
 
@@ -120,7 +111,16 @@ public void onResult(String result) {
     }
 
     @Override
-    public void onStats(float tps) {
-        tokensPerSecond.add(tps);
+    public void onStats(String stats) {
+        float tps = 0;
+        try {
+            JSONObject jsonObject = new JSONObject(stats);
+            int numGeneratedTokens = jsonObject.getInt("generated_tokens");
+            int inferenceEndMs = jsonObject.getInt("inference_end_ms");
+            int promptEvalEndMs = jsonObject.getInt("prompt_eval_end_ms");
+            tps = (float) numGeneratedTokens / (inferenceEndMs - promptEvalEndMs) * 1000;
+            tokensPerSecond.add(tps);
+        } catch (JSONException e) {
+        }
     }
 }