From 57779d7ea1606f00238d5405bd99dda54d657420 Mon Sep 17 00:00:00 2001 From: Sanghun Lee Date: Mon, 5 Jun 2023 21:37:33 +0900 Subject: [PATCH] fix: Handle buggy orm field loading when destroy session (#1312) Co-authored-by: Joongi Kim --- BUILD | 2 +- changes/1312.fix.md | 1 + src/ai/backend/manager/registry.py | 2 +- 3 files changed, 3 insertions(+), 2 deletions(-) create mode 100644 changes/1312.fix.md diff --git a/BUILD b/BUILD index c9bda93068..8d74b81436 100644 --- a/BUILD +++ b/BUILD @@ -23,7 +23,7 @@ python_requirements( "types-PyYAML": ["yaml"], "types-python-dateutil": ["dateutil", "dateutil.parser", "dateutil.tz"], "types-redis": ["redis"], - "types-six": ["six"], + "types-six": ["six", "graphql", "promise"], "types-tabulate": ["tabulate"], }, ) diff --git a/changes/1312.fix.md b/changes/1312.fix.md new file mode 100644 index 0000000000..ad5ee7ec5e --- /dev/null +++ b/changes/1312.fix.md @@ -0,0 +1 @@ +Handle buggy ORM field loading when destroy session. diff --git a/src/ai/backend/manager/registry.py b/src/ai/backend/manager/registry.py index 922d565c3e..7357741677 100644 --- a/src/ai/backend/manager/registry.py +++ b/src/ai/backend/manager/registry.py @@ -2223,7 +2223,7 @@ async def destroy_session( await self.event_producer.produce_event( SessionCancelledEvent( session_id, - kernel.session_creation_id, + target_session.creation_id, reason, ), )