From cf31207d75bf4a7224a45bdf66afb805abccf3c6 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Wed, 22 Apr 2026 09:16:23 +0000
Subject: [PATCH] Fix mmpose import error preservation

Agent-Logs-Url: https://github.com/IBB666/IBB_POSE/sessions/8173cd3d-9400-4422-8793-9248406eb79e

Co-authored-by: kukulu00 <1170651+kukulu00@users.noreply.github.com>
---
 models/HeatmapHead.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/models/HeatmapHead.py b/models/HeatmapHead.py
index 753499c..35ece2c 100644
--- a/models/HeatmapHead.py
+++ b/models/HeatmapHead.py
@@ -15,12 +15,13 @@ from types import SimpleNamespace
 import numpy as np
 from torch import nn
 
+_mmpose_import_error = None
+
 try:
     from mmpose.registry import MODELS
-except ImportError as _mmpose_import_error:
+except ImportError as exc:
     MODELS = None
-else:
-    _mmpose_import_error = None
+    _mmpose_import_error = exc
 
 
 def _build_head_cfg(mode="body"):