change entity type text
Browse files
utils.py
CHANGED
|
@@ -314,7 +314,7 @@ class PIIMasker:
|
|
| 314 |
new_text_parts.append(text[current_pos:entity.start])
|
| 315 |
|
| 316 |
# Add the mask
|
| 317 |
-
mask = f"[{entity.entity_type
|
| 318 |
new_text_parts.append(mask)
|
| 319 |
|
| 320 |
current_pos = entity.end
|
|
|
|
| 314 |
new_text_parts.append(text[current_pos:entity.start])
|
| 315 |
|
| 316 |
# Add the mask
|
| 317 |
+
mask = f"[{entity.entity_type}]" # Changed to upper for clarity
|
| 318 |
new_text_parts.append(mask)
|
| 319 |
|
| 320 |
current_pos = entity.end
|