Refactored TelemetrySender.send function

openvinotoolkit · Mar 18, 2021 · ac20ea6 · nosovmik · Mar 18, 2021 · ac20ea6
1 parent a330e91
commit ac20ea6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/model-optimizer/telemetry/utils/sender.py b/model-optimizer/telemetry/utils/sender.py
@@ -34,15 +34,15 @@ def _future_callback(future):
             with self._lock:
                 self.queue_size -= 1
 
-        callback_needed = False
+        fut = None
         with self._lock:
             if self.queue_size < MAX_QUEUE_SIZE:
                 fut = self.executor.submit(backend.send, message)
-                callback_needed = True
                 self.queue_size += 1
             else:
                 pass  # dropping a message because the queue is full
-        if callback_needed:
+        # to avoid dead lock we should not add callback inside the "with self._lock" block
+        if fut is not None:
             fut.add_done_callback(_future_callback)
 
     def force_shutdown(self, timeout: float):