Skip to content

Commit 46395bc

Browse files
committed
update stop multi
1 parent b5ef804 commit 46395bc

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

FlagEmbedding/abc/evaluation/searcher.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ def __str__(self) -> str:
2828
def stop_multi_process_pool(self):
2929
if self.embedder.pool is not None:
3030
self.embedder.stop_multi_process_pool(self.embedder.pool)
31+
self.embedder.pool = None
3132

3233
@abstractmethod
3334
def __call__(
@@ -151,6 +152,7 @@ def __str__(self) -> str:
151152
def stop_multi_process_pool(self):
152153
if self.reranker.pool is not None:
153154
self.reranker.stop_multi_process_pool(self.reranker.pool)
155+
self.reranker.pool = None
154156

155157
def __call__(
156158
self,

0 commit comments

Comments
 (0)