Update some PYRE overrides

Author: mthrok

examples/benchmark_utils.py

@@ -197,7 +197,8 @@ def _warmup_executor(
     last_output: T | None = None
     for future in as_completed(futures):
         last_output = future.result()
-    return last_output  # pyre-ignore[7]
+    assert last_output is not None
+    return last_output
 
 
 class BenchmarkRunner:
@@ -282,7 +283,8 @@ def _run_iterations(
             qps_samples.append(iterations / elapsed)
             cpu_samples.append(cpu_percent / iterations)
 
-        return qps_samples, cpu_samples, last_output  # pyre-ignore[7]
+        assert last_output is not None
+        return qps_samples, cpu_samples, last_output
 
     def run(
         self,

src/spdl/_internal/import_utils.py

@@ -4,10 +4,12 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-# pyre-unsafe
+# pyre-strict
 
 import importlib
+from collections.abc import Callable
 from types import ModuleType
+from typing import Any
 
 __all__ = [
     "lazy_import",
@@ -17,28 +19,28 @@
 class _LazilyImportedModule(ModuleType):
     """Delay module import until its attribute is accessed."""
 
-    def __init__(self, name, import_func):
+    def __init__(self, name: str, import_func: Callable[[], ModuleType]) -> None:
         super().__init__(name)
         self.import_func = import_func
-        self.module = None
+        self.module: ModuleType | None = None
 
     # Note:
     # Python caches what was retrieved with `__getattr__`, so this method will not be
     # called again for the same item.
-    def __getattr__(self, item):
+    def __getattr__(self, item: str) -> Any:
         self._import_once()
         return getattr(self.module, item)
 
-    def __repr__(self):
+    def __repr__(self) -> str:
         if self.module is None:
             return f"<module '{self.__module__}.{self.__class__.__name__}(\"{self.__name__}\")'>"
         return repr(self.module)
 
-    def __dir__(self):
+    def __dir__(self) -> list[str]:
         self._import_once()
         return dir(self.module)
 
-    def _import_once(self):
+    def _import_once(self) -> None:
         if self.module is None:
             self.module = self.import_func()
             # Note:

tests/io/configs_test.py

@@ -4,7 +4,7 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-# pyre-unsafe
+# pyre-strict
 
 import unittest
 

tests/io/frames_test.py

@@ -4,7 +4,7 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-# pyre-unsafe
+# pyre-strict
 
 import unittest
 

tests/io/zero_copy_bytes_passing_test.py

@@ -4,25 +4,27 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-# pyre-unsafe
+# pyre-strict
 
 import unittest
+from typing import Any
 
 import numpy as np
 import spdl.io
+from numpy.typing import NDArray
 
 from ..fixture import FFMPEG_CLI, get_sample
 
 
-def _decode(media_type, src):
+def _decode(media_type: str, src: str | bytes) -> NDArray[Any]:
     demux_func = {
         "audio": spdl.io.demux_audio,
         "video": spdl.io.demux_video,
         "image": spdl.io.demux_image,
     }[media_type]
 
     packets = demux_func(src)
-    frames = spdl.io.decode_packets(packets)
+    frames = spdl.io.decode_packets(packets)  # pyre-ignore[6]
     buffer = spdl.io.convert_frames(frames)
     return spdl.io.to_numpy(buffer)
 

tests/pipeline/pipeline_failure_exceptstar_test.py

@@ -4,7 +4,7 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-# pyre-unsafe
+# pyre-strict
 
 """Tests for PipelineFailure using ``except*`` syntax (Python 3.11+ only).
 
@@ -15,20 +15,21 @@
 
 import sys
 import unittest
+from collections.abc import Iterator
 
 if sys.version_info < (3, 11):
     raise unittest.SkipTest("except* syntax requires Python 3.11+")
 
 from spdl.pipeline import PipelineBuilder
 
 
-def passthrough(x):
+def passthrough(x: int) -> int:
     return x
 
 
 class TestPipelineFailureExceptStar(unittest.TestCase):
     def test_pipeline_failure_except_star(self) -> None:
-        def failing_range(n):
+        def failing_range(n: int) -> Iterator[int]:
             yield from range(n)
             raise ValueError("Iterator failed")