From 030739b810cec79a482c4fc92f543c37fa6448bb Mon Sep 17 00:00:00 2001 From: thomas chaton Date: Fri, 25 Nov 2022 10:30:01 +0000 Subject: [PATCH 1/2] update --- src/lightning_app/utilities/proxies.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/lightning_app/utilities/proxies.py b/src/lightning_app/utilities/proxies.py index 07b03da7d9201..2e7dc6ce7efec 100644 --- a/src/lightning_app/utilities/proxies.py +++ b/src/lightning_app/utilities/proxies.py @@ -17,6 +17,7 @@ from deepdiff import DeepDiff, Delta from lightning_utilities.core.apply_func import apply_to_collection +from lightning_app.core.queues import MultiProcessQueue from lightning_app.storage import Path from lightning_app.storage.copier import _Copier, _copy_files from lightning_app.storage.drive import _maybe_create_drive, Drive @@ -357,7 +358,7 @@ def enable_spawn(self) -> Generator: yield def _clean_queues(self): - if "LIGHTNING_APP_STATE_URL" in os.environ: + if not isinstance(self.work._request_queue, MultiProcessQueue): self.work._request_queue = self.work._request_queue.to_dict() self.work._response_queue = self.work._response_queue.to_dict() From 581647fd9904a28b02bec19b3491274b21bd1915 Mon Sep 17 00:00:00 2001 From: thomas chaton Date: Fri, 25 Nov 2022 10:31:49 +0000 Subject: [PATCH 2/2] update --- src/lightning_app/CHANGELOG.md | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/lightning_app/CHANGELOG.md b/src/lightning_app/CHANGELOG.md index 305ee591b0257..39b06d2d1a22a 100644 --- a/src/lightning_app/CHANGELOG.md +++ b/src/lightning_app/CHANGELOG.md @@ -41,6 +41,9 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/). - Fixed the work not stopped when successful when passed directly to the LightningApp ([#15801](https://github.com/Lightning-AI/lightning/pull/15801)) +- Fixed the `enable_spawn` method of the `WorkRunExecutor` ([#15812](https://github.com/Lightning-AI/lightning/pull/15812)) + + ## [1.8.2] - 2022-11-17 ### Added