From 57604c3c70d75be0fd88bd0661adc9822460c364 Mon Sep 17 00:00:00 2001
From: Kyujin Cho <kyujin.cho@lablup.com>
Date: Sun, 22 Dec 2024 16:54:33 +0900
Subject: [PATCH] fix: 'DockerKernel' object has no attribute 'network_driver'

---
 src/ai/backend/agent/docker/kernel.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/ai/backend/agent/docker/kernel.py b/src/ai/backend/agent/docker/kernel.py
index edd91122aa..427ca6cfa3 100644
--- a/src/ai/backend/agent/docker/kernel.py
+++ b/src/ai/backend/agent/docker/kernel.py
@@ -83,6 +83,8 @@ def __getstate__(self):
         return props
 
     def __setstate__(self, props):
+        if "network_driver" not in props:
+            props["network_driver"] = "bridge"
         super().__setstate__(props)
 
     async def create_code_runner(