From acc83367322ba793c4851a841b81aed310a2ecc2 Mon Sep 17 00:00:00 2001
From: Frank Liu <frankfliu2000@gmail.com>
Date: Tue, 1 Oct 2024 09:28:12 -0700
Subject: [PATCH] [api] alternative NDArray should not be closed in NDScope

---
 api/src/main/java/ai/djl/ndarray/NDArrayAdapter.java | 1 +
 1 file changed, 1 insertion(+)

diff --git a/api/src/main/java/ai/djl/ndarray/NDArrayAdapter.java b/api/src/main/java/ai/djl/ndarray/NDArrayAdapter.java
index 04ea056b4db..cceb5efd494 100644
--- a/api/src/main/java/ai/djl/ndarray/NDArrayAdapter.java
+++ b/api/src/main/java/ai/djl/ndarray/NDArrayAdapter.java
@@ -1346,6 +1346,7 @@ private NDArray getAlternativeArray() {
         } else {
             alternativeArray.set(getDataType().asDataType(toByteBuffer()));
         }
+        NDScope.unregister(alternativeArray);
         return alternativeArray;
     }
 }