wool 0.1rc10__tar.gz → 0.1rc11__tar.gz

{wool-0.1rc10 → wool-0.1rc11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wool
-Version: 0.1rc10
+Version: 0.1rc11
 Summary: A Python framework for distributed multiprocessing.
 Author-email: Conrad Bzura <conrad@wool.io>
 Maintainer-email: maintainers@wool.io

{wool-0.1rc10 → wool-0.1rc11}/wool/_protobuf/task_pb2_grpc.py

@@ -4,7 +4,7 @@ import grpc
 import warnings
 
 
-GRPC_GENERATED_VERSION = '1.75.0'
+GRPC_GENERATED_VERSION = '1.75.1'
 GRPC_VERSION = grpc.__version__
 _version_not_supported = False
 

{wool-0.1rc10 → wool-0.1rc11}/wool/_protobuf/worker_pb2_grpc.py

@@ -6,7 +6,7 @@ import warnings
 from . import task_pb2 as task__pb2
 from . import worker_pb2 as worker__pb2
 
-GRPC_GENERATED_VERSION = '1.75.0'
+GRPC_GENERATED_VERSION = '1.75.1'
 GRPC_VERSION = grpc.__version__
 _version_not_supported = False
 

{wool-0.1rc10 → wool-0.1rc11}/wool/_worker.py

@@ -334,20 +334,20 @@ class LocalWorker(Worker[_T_RegistryService]):
         the registry service. If graceful shutdown fails, the process
         is forcefully terminated.
         """
-        if not self._info:
-            raise RuntimeError("Cannot unregister - worker has no info")
-
-        await self._registry_service.unregister(self._info)
-
-        if not self._worker_process.is_alive():
-            return
         try:
-            if self._worker_process.pid:
-                os.kill(self._worker_process.pid, signal.SIGINT)
-                self._worker_process.join()
-        except OSError:
-            if self._worker_process.is_alive():
-                self._worker_process.kill()
+            if not self._info:
+                raise RuntimeError("Cannot unregister - worker has no info")
+            await self._registry_service.unregister(self._info)
+        finally:
+            if not self._worker_process.is_alive():
+                return
+            try:
+                if self._worker_process.pid:
+                    os.kill(self._worker_process.pid, signal.SIGINT)
+                    self._worker_process.join()
+            except OSError:
+                if self._worker_process.is_alive():
+                    self._worker_process.kill()
 
 
 class WorkerProcess(Process):

{wool-0.1rc10 → wool-0.1rc11}/wool/_worker_pool.py

@@ -312,7 +312,7 @@ class WorkerPool:
         self, uri, *tags: str, size: int, factory: WorkerFactory | None
     ):
         if factory is None:
-            factory = partial(LocalWorker, registry_service=LocalRegistryService(uri))
+            factory = self._default_worker_factory(uri)
 
         tasks = []
         for _ in range(size):
@@ -329,3 +329,9 @@ class WorkerPool:
         """Sends a stop command to all workers and unregisters them."""
         tasks = [asyncio.create_task(worker.stop()) for worker in self._workers]
         await asyncio.gather(*tasks, return_exceptions=True)
+
+    def _default_worker_factory(self, uri):
+        def factory(*tags, **_):
+            return LocalWorker(*tags, registry_service=LocalRegistryService(uri))
+
+        return factory