From 356015896a1fbf74a590c4012ea3e3858db054cb Mon Sep 17 00:00:00 2001 From: Jianhua Zheng Date: Wed, 29 Nov 2023 02:47:07 +0000 Subject: [PATCH] skip omegaconf to avoid stack overflow due to recursive calls --- src/onediff/infer_compiler/transform/manager.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/onediff/infer_compiler/transform/manager.py b/src/onediff/infer_compiler/transform/manager.py index f2890caf2..dc7976f0a 100644 --- a/src/onediff/infer_compiler/transform/manager.py +++ b/src/onediff/infer_compiler/transform/manager.py @@ -86,6 +86,8 @@ def update_class_proxies(self, class_proxy_dict: Dict[str, type], verbose=True): self.logger.debug(debug_message) def _transform_entity(self, entity): + if "omegaconf" in str(entity): + return entity result = self.mocker.mock_entity(entity) if result is None: RuntimeError(f"Failed to transform entity: {entity}")