Fix for older proto generators not having ValueType

modal-labs · Nov 19, 2024 · 4728708 · 4728708
1 parent 82a4648
commit 4728708
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/modal/io_streams.py b/modal/io_streams.py
@@ -30,7 +30,7 @@
 
 
 async def _sandbox_logs_iterator(
-    sandbox_id: str, file_descriptor: api_pb2.FileDescriptor.ValueType, last_entry_id: str, client: _Client
+    sandbox_id: str, file_descriptor: "api_pb2.FileDescriptor.ValueType", last_entry_id: str, client: _Client
 ) -> AsyncGenerator[Tuple[Optional[bytes], str], None]:
     req = api_pb2.SandboxGetLogsRequest(
         sandbox_id=sandbox_id,
@@ -49,7 +49,7 @@ async def _sandbox_logs_iterator(
 
 
 async def _container_process_logs_iterator(
-    process_id: str, file_descriptor: api_pb2.FileDescriptor.ValueType, client: _Client
+    process_id: str, file_descriptor: "api_pb2.FileDescriptor.ValueType", client: _Client
 ) -> AsyncGenerator[Optional[bytes], None]:
     req = api_pb2.ContainerExecGetOutputRequest(
         exec_id=process_id, timeout=55, file_descriptor=file_descriptor, get_raw_bytes=True
@@ -90,7 +90,7 @@ class _StreamReader(Generic[T]):
 
     def __init__(
         self,
-        file_descriptor: api_pb2.FileDescriptor.ValueType,
+        file_descriptor: "api_pb2.FileDescriptor.ValueType",
         object_id: str,
         object_type: Literal["sandbox", "container_process"],
         client: _Client,