fix: 'DockerKernel' object has no attribute 'network_driver'

lablup · Dec 22, 2024 · 57604c3 · 57604c3
1 parent 42b4030
commit 57604c3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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(