Fix type error in test_python_actors.py (#1393)

Summary:
Pull Request resolved: #1393

This fixes broken github CI due to `ProcessAllocator` being marked final.
ghstack-source-id: 313483343
exported-using-ghexport

Reviewed By: benjipelletier

Differential Revision: D83671339

fbshipit-source-id: dbccfb3626824200416451d6568ce7f746b079e8

Author: samlurye

python/monarch/_rust_bindings/monarch_hyperactor/alloc.pyi

@@ -10,8 +10,6 @@ from datetime import timedelta
 
 from typing import Dict, final, Optional
 
-from monarch._rust_bindings.monarch_hyperactor.alloc import Alloc, AllocSpec
-from monarch._rust_bindings.monarch_hyperactor.mailbox import Mailbox
 from monarch._rust_bindings.monarch_hyperactor.pytokio import PythonTask
 
 from typing_extensions import Self

python/monarch/_src/actor/allocator.py

@@ -102,7 +102,6 @@ def _stream_logs(self) -> bool:
         ...
 
 
-@final
 class ProcessAllocator(ProcessAllocatorBase, AllocateMixin):
     """
     An allocator that allocates by spawning local processes.

python/tests/test_python_actors.py

@@ -30,6 +30,7 @@
     PythonMessage,
     PythonMessageKind,
 )
+from monarch._rust_bindings.monarch_hyperactor.alloc import Alloc, AllocSpec
 from monarch._rust_bindings.monarch_hyperactor.mailbox import (
     PortId,
     PortRef,
@@ -790,6 +791,11 @@ async def test_stream_logs_case(stream_logs: bool, test_name: str) -> None:
                     else:
 
                         class ProcessAllocatorStreamLogs(ProcessAllocator):
+                            def allocate_nonblocking(
+                                self, spec: AllocSpec
+                            ) -> PythonTask[Alloc]:
+                                return super().allocate_nonblocking(spec)
+
                             def _stream_logs(self) -> bool:
                                 return stream_logs