fix: Prevent the occasional unhandled UnboundLocalError in agent st…

…ats routine - Skip measuring the stat for an agent registry item if it has not yet assigned container ID
lablup · Jun 22, 2022 · 772bea7 · 772bea7
1 parent 96033aa
commit 772bea7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/changes/478.fix b/changes/478.fix
@@ -0,0 +1 @@
+Skip measuring the stat for an agent registry item if it has not yet assigned container ID to prevent the occasional unhandled `UnboundLocalError`.
diff --git a/src/ai/backend/agent/stats.py b/src/ai/backend/agent/stats.py
@@ -366,7 +366,8 @@ async def collect_container_stat(
                     cid = info['container_id']
                 except KeyError:
                     log.warning('collect_container_stat(): no container for kernel {}', kid)
-                kernel_id_map[ContainerId(cid)] = kid
+                else:
+                    kernel_id_map[ContainerId(cid)] = kid
             unused_kernel_ids = set(self.kernel_metrics.keys()) - set(kernel_id_map.values())
             for unused_kernel_id in unused_kernel_ids:
                 log.debug('removing kernel_metric for {}', unused_kernel_id)