Fixup for handling the sse-keepalive option in the CLI (from #9722)

Scille · Feb 17, 2025 · ac76118 · ac76118
1 parent 6df137a
commit ac76118
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/server/parsec/cli/run.py b/server/parsec/cli/run.py
@@ -2,7 +2,6 @@
 from __future__ import annotations
 
 import asyncio
-import math
 import tempfile
 from pathlib import Path
 from typing import Any
@@ -323,8 +322,8 @@ def handle_parse_result(
     "--sse-keepalive",
     default=30,
     show_default=True,
-    type=float,
-    callback=lambda ctx, param, value: math.inf if value is None or value <= 0 else value,
+    type=int,
+    callback=lambda ctx, param, value: None if value is None or value <= 0 else value,
     envvar="PARSEC_SSE_KEEPALIVE",
     show_envvar=True,
     help="Keep SSE connection open by sending keepalive messages to client, in seconds (pass <= 0 to disable)",