[python] Fixes python startup race condition

deepjavalibrary · Mar 30, 2023 · e46521d · e46521d
1 parent 77bf5ad
commit e46521d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/engines/python/src/main/java/ai/djl/python/engine/PyProcess.java b/engines/python/src/main/java/ai/djl/python/engine/PyProcess.java
@@ -205,14 +205,16 @@ synchronized void stopPythonProcess() {
 
     void setStarted(boolean started, int id) {
         if (restartCount.get() == id) {
+            this.started = started;
             if (started) {
                 latch.countDown();
-                this.started = latch.getCount() == 0;
             } else {
                 while (latch.getCount() > 0) {
                     latch.countDown();
                 }
             }
+        } else {
+            logger.warn("Unexpected restartCount: {}, expect: {}", id, restartCount.get());
         }
     }