From cbd3be039dd25cbb507c467d32df611aaf513945 Mon Sep 17 00:00:00 2001 From: beanbun Date: Mon, 13 Apr 2026 19:58:10 +0800 Subject: [PATCH 1/2] fix: Change the replacement symbol for masked entity --- graphgen/models/generator/masked_fill_in_blank_generator.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/graphgen/models/generator/masked_fill_in_blank_generator.py b/graphgen/models/generator/masked_fill_in_blank_generator.py index bc6f06c8..60672b0e 100644 --- a/graphgen/models/generator/masked_fill_in_blank_generator.py +++ b/graphgen/models/generator/masked_fill_in_blank_generator.py @@ -115,7 +115,8 @@ async def generate( match = re.search(mask_pattern, context) if match: gth = match.group(0) - masked_context = mask_pattern.sub("___", context) + # masked_context = mask_pattern.sub("___", context) + masked_context = mask_pattern.sub("{ }", context) else: logger.debug( "Regex Match Failed!\n" From 07af393adb9a11fae5ad49f91f13fd0837d5b444 Mon Sep 17 00:00:00 2001 From: chenzihong <58508660+ChenZiHong-Gavin@users.noreply.github.com> Date: Mon, 13 Apr 2026 20:09:09 +0800 Subject: [PATCH 2/2] Apply suggestion from @gemini-code-assist[bot] Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> --- graphgen/models/generator/masked_fill_in_blank_generator.py | 1 - 1 file changed, 1 deletion(-) diff --git a/graphgen/models/generator/masked_fill_in_blank_generator.py b/graphgen/models/generator/masked_fill_in_blank_generator.py index 60672b0e..355cadcf 100644 --- a/graphgen/models/generator/masked_fill_in_blank_generator.py +++ b/graphgen/models/generator/masked_fill_in_blank_generator.py @@ -115,7 +115,6 @@ async def generate( match = re.search(mask_pattern, context) if match: gth = match.group(0) - # masked_context = mask_pattern.sub("___", context) masked_context = mask_pattern.sub("{ }", context) else: logger.debug(