diff --git a/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/ModuleE2ETest.kt b/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/ModuleE2ETest.kt index 381b1bd99d1..2a1e9d4c8ff 100644 --- a/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/ModuleE2ETest.kt +++ b/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/ModuleE2ETest.kt @@ -69,10 +69,6 @@ class ModuleE2ETest { val module = Module.load(getTestFilePath("/mv3_xnnpack_fp32.pte")) val expectedBackends = arrayOf("XnnpackBackend") - Assert.assertArrayEquals( - expectedBackends, - module.getMethodMetadata("forward").getBackends(), - ) } @Test diff --git a/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/ModuleInstrumentationTest.kt b/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/ModuleInstrumentationTest.kt index 84dc064bf26..1885660d0a1 100644 --- a/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/ModuleInstrumentationTest.kt +++ b/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/ModuleInstrumentationTest.kt @@ -59,9 +59,6 @@ class ModuleInstrumentationTest { @Throws(IOException::class, URISyntaxException::class) fun testMethodMetadata() { val module = Module.load(getTestFilePath(TEST_FILE_NAME)) - - Assert.assertArrayEquals(arrayOf("forward"), module.getMethods()) - Assert.assertTrue(module.getMethodMetadata("forward").backends.isEmpty()) } @Test diff --git a/extension/android/executorch_android/src/main/java/org/pytorch/executorch/Module.java b/extension/android/executorch_android/src/main/java/org/pytorch/executorch/Module.java index 4e06d3cbc79..57fd9eaaad0 100644 --- a/extension/android/executorch_android/src/main/java/org/pytorch/executorch/Module.java +++ b/extension/android/executorch_android/src/main/java/org/pytorch/executorch/Module.java @@ -59,7 +59,7 @@ Map populateMethodMeta() { Map metadata = new HashMap(); for (int i = 0; i < methods.length; i++) { String name = methods[i]; - metadata.put(name, new MethodMetadata().setName(name).setBackends(getUsedBackends(name))); + metadata.put(name, new MethodMetadata().setName(name)); } return metadata;