diff --git a/tester/api_config/config_analyzer.py b/tester/api_config/config_analyzer.py index 0b83fec9..2cc2a2f4 100644 --- a/tester/api_config/config_analyzer.py +++ b/tester/api_config/config_analyzer.py @@ -2809,6 +2809,13 @@ def __init__(self, config): self.config = config self.args = [] self.kwargs = collections.OrderedDict() + + # 兼容 paddle.Size([...]) 格式:将其替换为 [...] + def replace_paddle_size(match): + shape_list = match.group(1) # 提取 [...] 部分 + return shape_list + + config = re.sub(r"paddle\.Size\(\s*(\[[^\]]*\])\s*\)", replace_paddle_size, config) config = config.replace("Tensor(", "TensorConfig(") self.api_name, offset = self.get_api(config)