type

modal-labs · Jul 3, 2024 · cdabb79 · cdabb79
1 parent 7021091
commit cdabb79
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/modal/_utils/blob_utils.py b/modal/_utils/blob_utils.py
@@ -327,7 +327,7 @@ async def blob_iter(blob_id, stub) -> AsyncIterator[bytes]:
 
 @dataclasses.dataclass
 class FileUploadSpec:
-    source: Callable[[], BinaryIO]
+    source: Callable[[], Union[_GeneratorContextManager, BinaryIO]]
     source_description: Any
     mount_filename: str
 
@@ -339,7 +339,7 @@ class FileUploadSpec:
 
 
 def _get_file_upload_spec(
-    source: Callable[[], _GeneratorContextManager[Any]],
+    source: Callable[[], Union[_GeneratorContextManager, BinaryIO]],
     source_description: Any,
     mount_filename: PurePosixPath,
     mode: int,