Moving the finally block.

microsoft · Nov 8, 2023 · 796e4b1 · 796e4b1
1 parent 437b52e
commit 796e4b1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/...native/android/src/androidTest/java/ai/onnxruntime/reactnative/OnnxruntimeModuleTest.java b/...native/android/src/androidTest/java/ai/onnxruntime/reactnative/OnnxruntimeModuleTest.java
@@ -254,9 +254,10 @@ public void throwWrongSizeInput() {
         Assert.fail("Should have thrown exception");
       } catch (Exception e) {
         Assert.assertTrue(e.getMessage().contains("Got invalid dimensions for input"));
+      } finally {
+        ortModule.dispose(sessionKey);
       }
     } finally {
-      ortModule.dispose(sessionKey);
       mockSession.finishMocking();
     }
   }
@@ -313,9 +314,10 @@ public void throwWrongRankInput() {
         Assert.fail("Should have thrown exception");
       } catch (Exception e) {
         Assert.assertTrue(e.getMessage().contains("Invalid rank for input"));
+      } finally {
+        ortModule.dispose(sessionKey);
       }
     } finally {
-      ortModule.dispose(sessionKey);
       mockSession.finishMocking();
     }
   }