fbcode/spdl/examples/image_dataloading.py

Differential Revision: D110967070

Pull Request resolved: #1600

Author: frbng

examples/image_dataloading.py

@@ -32,6 +32,8 @@
 
 # pyre-strict
 
+from __future__ import annotations
+
 import argparse
 import logging
 import signal
@@ -42,6 +44,7 @@
 from functools import partial
 from pathlib import Path
 from threading import Event
+from types import FrameType
 
 import spdl.io
 import spdl.io.utils
@@ -222,7 +225,7 @@ def worker_entrypoint(args_: list[str]) -> PerfResult:
 
     ev: Event = Event()
 
-    def handler_stop_signals(_signum, _frame) -> None:
+    def handler_stop_signals(_signum: int, _frame: FrameType | None) -> None:
         ev.set()
 
     signal.signal(signal.SIGTERM, handler_stop_signals)