fix client ready gpus returning bug

Spico197 · Jul 13, 2021 · c953bb2 · c953bb2
1 parent 010690f
commit c953bb2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/watchmen/client.py b/watchmen/client.py
@@ -124,7 +124,7 @@ def ping(self):
             if result["msg"] == ClientStatus.WAITING:
                 return False, result["available_gpus"]
             elif result["msg"] == ClientStatus.READY:
-                return False, result["available_gpus"]
+                return True, result["available_gpus"]
             elif result["msg"] == ClientStatus.OK:
                 logger.warning("Status is OK, which has finished requesting GPUs.")
                 return False, result["available_gpus"]