More guard

Signed-off-by: Joey Yu <[email protected]>
awsonbudget · Apr 26, 2023 · 8e02721 · 8e02721
1 parent 7516cfd
commit 8e02721
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/internal/worker.py b/src/internal/worker.py
@@ -6,12 +6,12 @@
 
 from src.internal.manager import Location
 from src.internal.type import Status, WsType
-from src.utils.config import manager, cluster_group, etcd_client
+from src.utils.config import manager, cluster_group, etcd_client, is_prod
 from src.utils.ws import update
 
 
 async def state_saver():
-    while True:
+    while is_prod:
         await asyncio.sleep(10)
         serialized_manager = pickle.dumps(manager)
         etcd_client.put("manager", serialized_manager)