ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-03 02:32:11 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix shutting down a Task immediately after creation might block

Browse Source
This commit is contained in:
allegroai 2021-05-14 09:42:42 +03:00
parent 899ff93a73
commit 350a39b886
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
clearml/utilities/process/mp.py
View File

@ -61,8 +61,9 @@ class SafeQueue(object):
return self._q.empty() and (not self._internal_q or self._internal_q.empty())
def is_pending(self):
# check if we have pending requests to be pushed (it does not mean they were pulled)
# only call from main put process
return self._q_size > 0 or not self.empty()
return self._q_size > 0
def close(self, event):
# wait until all pending requests pushed