From 07574fddbb2cd74cb4e9368ac46e5f6084396533 Mon Sep 17 00:00:00 2001
From: Tong Gao <gaotongxiao@gmail.com>
Date: Fri, 22 Sep 2023 17:30:54 +0800
Subject: [PATCH] [Fix] keep keys (#431)

---
 opencompass/partitioners/base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/opencompass/partitioners/base.py b/opencompass/partitioners/base.py
index cd523faa..c6015bbd 100644
--- a/opencompass/partitioners/base.py
+++ b/opencompass/partitioners/base.py
@@ -62,7 +62,7 @@ class BasePartitioner:
                         tgt_ptr[key] = {}
                     tgt_ptr = tgt_ptr[key]
                 tgt_ptr[key_chain[-1]] = ori_ptr[key_chain[-1]]
-            except AttributeError:
+            except Exception:
                 self.logger.warning(f'Key {k} not found in config, ignored.')
 
         tasks = self.partition(models,