fix: convert environment variables to int (#70)

* convert environment variables to int COMFY_POLLING_INTERVAL_MS and COMFY_POLLING_MAX_RETRIES should be integers. * chore: remove unnecessary comment --------- Co-authored-by: NERDDISCO <[email protected]>
blib-la · Nov 10, 2024 · 7ab3d2a · 7ab3d2a
1 parent caf8478
commit 7ab3d2a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/rp_handler.py b/src/rp_handler.py
@@ -14,9 +14,9 @@
 # Maximum number of API check attempts
 COMFY_API_AVAILABLE_MAX_RETRIES = 500
 # Time to wait between poll attempts in milliseconds
-COMFY_POLLING_INTERVAL_MS = os.environ.get("COMFY_POLLING_INTERVAL_MS", 250)
+COMFY_POLLING_INTERVAL_MS = int(os.environ.get("COMFY_POLLING_INTERVAL_MS", 250))
 # Maximum number of poll attempts
-COMFY_POLLING_MAX_RETRIES = os.environ.get("COMFY_POLLING_MAX_RETRIES", 500)
+COMFY_POLLING_MAX_RETRIES = int(os.environ.get("COMFY_POLLING_MAX_RETRIES", 500))
 # Host where ComfyUI is running
 COMFY_HOST = "127.0.0.1:8188"
 # Enforce a clean state after each job is done