[Fix] keep keys (#431)

open-compass · Sep 22, 2023 · 07574fd · 07574fd
1 parent a1ea3c0
commit 07574fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/opencompass/partitioners/base.py b/opencompass/partitioners/base.py
@@ -62,7 +62,7 @@ def __call__(self, cfg: ConfigDict) -> List[Dict]:
                         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,