From a12e32f4afab5dbf227b90f0c999624c542c555b Mon Sep 17 00:00:00 2001
From: Sanghun Lee <sanghun@lablup.com>
Date: Tue, 11 Feb 2025 19:54:21 +0900
Subject: [PATCH] fix(BA-694): Number of concurrent sftp session queried
 incorrectly (#3654)

---
 changes/3654.fix.md                     | 1 +
 src/ai/backend/manager/models/kernel.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)
 create mode 100644 changes/3654.fix.md

diff --git a/changes/3654.fix.md b/changes/3654.fix.md
new file mode 100644
index 00000000000..0631dcd4fe0
--- /dev/null
+++ b/changes/3654.fix.md
@@ -0,0 +1 @@
+Correct the number of concurrent SFTP sessions queried from DB
diff --git a/src/ai/backend/manager/models/kernel.py b/src/ai/backend/manager/models/kernel.py
index 60f57bf3c08..e42355e20be 100644
--- a/src/ai/backend/manager/models/kernel.py
+++ b/src/ai/backend/manager/models/kernel.py
@@ -1609,7 +1609,7 @@ async def recalc_concurrency_used(
             .where(
                 (KernelRow.access_key == access_key)
                 & (KernelRow.status.in_(USER_RESOURCE_OCCUPYING_KERNEL_STATUSES))
-                & (KernelRow.session_type.not_in(PRIVATE_SESSION_TYPES))
+                & (KernelRow.session_type.in_(PRIVATE_SESSION_TYPES))
             ),
         )
         sftp_concurrency_used = result.scalar()