Skip to content

Commit e52bdf0

Browse files
superfartherbeanbunChenZiHong-Gavingemini-code-assist[bot]
authored
fix: Change the replacement symbol for masked entity (#188)
* fix: Change the replacement symbol for masked entity * Apply suggestion from @gemini-code-assist[bot] Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> --------- Co-authored-by: beanbun <yuanzhonghang@pjlab.org.cn> Co-authored-by: chenzihong <58508660+ChenZiHong-Gavin@users.noreply.github.com> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
1 parent 83f55b2 commit e52bdf0

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

graphgen/models/generator/masked_fill_in_blank_generator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ async def generate(
115115
match = re.search(mask_pattern, context)
116116
if match:
117117
gth = match.group(0)
118-
masked_context = mask_pattern.sub("___", context)
118+
masked_context = mask_pattern.sub("{ }", context)
119119
else:
120120
logger.debug(
121121
"Regex Match Failed!\n"

0 commit comments

Comments
 (0)