From 87ac7c6a82cc7d174075ab79934d63332546b8f3 Mon Sep 17 00:00:00 2001
From: carefree0910 <syameimaru.saki@gmail.com>
Date: Fri, 18 Oct 2024 16:32:56 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9BImplemented=20`=5F=5Fdel=5F=5F`=20f?=
 =?UTF-8?q?or=20`AsyncDataLoaderIter`?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 core/learn/schema.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/core/learn/schema.py b/core/learn/schema.py
index 86b0916..a0b7fdf 100644
--- a/core/learn/schema.py
+++ b/core/learn/schema.py
@@ -262,6 +262,7 @@ class AsyncDataLoaderIter(_SingleProcessDataLoaderIter):
     _drained: bool
     _queue_cursor: int
     _dataset: IAsyncDataset
+    _finalized: bool
 
     def __init__(self, loader: "DataLoader"):
         super().__init__(loader)
@@ -271,17 +272,24 @@ def __init__(self, loader: "DataLoader"):
             raise RuntimeError(
                 "async prefetch is only available for `IAsyncDataset`"
             )  # pragma: no cover
+        self._finalized = True
         self._initialized = False
 
+    def __del__(self) -> None:
+        if not self._finalized:
+            self._dataset.async_finalize()
+
     def _initialize(self) -> None:
         self._queue = None
         self._drained = False
         self._queue_cursor = 0
         self._dataset.async_reset()
+        self._finalized = False
         self._initialized = True
 
     def _finalize(self) -> None:
         self._dataset.async_finalize()
+        self._finalized = True
         raise StopIteration
 
     def _sumbit_next(self) -> None: