fix: mypy

renan-r-santos · Apr 17, 2024 · da07699 · da07699
1 parent b87a759
commit da07699
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/pixi_kernel/__main__.py b/src/pixi_kernel/__main__.py
@@ -6,6 +6,7 @@
 import sys
 import typing
 from pathlib import Path
+from types import FrameType
 
 from ipykernel.ipkernel import IPythonKernel
 from ipykernel.kernelapp import IPKernelApp
@@ -64,16 +65,17 @@ def main() -> None:
 
     logger.info(f"launching {kernel_display_name} kernel with {args}")
 
-    if platform.system() == "Windows":
+    if sys.platform == "win32":
         process = subprocess.Popen(args, env=env)
 
         forward_signals = set(signal.Signals) - {
             signal.CTRL_BREAK_EVENT,
             signal.CTRL_C_EVENT,
+            signal.SIGBREAK,
             signal.SIGTERM,
         }
 
-        def handle_signal(sig, frame):
+        def handle_signal(sig: int, frame: FrameType | None) -> None:
             process.send_signal(sig)
 
         for sig in forward_signals: