Moving the finally block.

microsoft · Oct 14, 2023 · a3c4d79 · a3c4d79
1 parent 3ab7e08
commit a3c4d79
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();
     }
   }