Skip to content

Commit cbd3be0

Browse files
author
beanbun
committed
fix: Change the replacement symbol for masked entity
1 parent 83f55b2 commit cbd3be0

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

graphgen/models/generator/masked_fill_in_blank_generator.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,8 @@ 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)
119+
masked_context = mask_pattern.sub("{ }", context)
119120
else:
120121
logger.debug(
121122
"Regex Match Failed!\n"

0 commit comments

Comments
 (0)