fix return type

pdm-project · Jun 2, 2023 · 3e2ba47 · 3e2ba47
1 parent 68e5b56
commit 3e2ba47
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/pdm/models/caches.py b/src/pdm/models/caches.py
@@ -3,7 +3,6 @@
 import contextlib
 import dataclasses
 import hashlib
-import io
 import json
 import os
 from functools import lru_cache
@@ -273,10 +272,12 @@ def get(self, key: str) -> bytes | None:
 
         return None
 
-    def get_body(self, key) -> io.FileIO | None:
+    def get_body(self, key: str) -> BinaryIO | None:
         path = self._get_cache_path(key)
         with contextlib.suppress(OSError):
-            return open(f"{path}.body", "rb")
+            return cast(BinaryIO, open(f"{path}.body", "rb"))
+
+        return None
 
     def set(self, key: str, value: bytes, expires: int | None = None) -> None:
         path = self._get_cache_path(key)