marcel1997 commited on
Commit
706eab0
·
verified ·
1 Parent(s): 5e26f4f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -251,7 +251,7 @@ def select_entities_based_on_json(prediction_json, entity_json):
251
  entities = []
252
  for key, value in prediction_json.items():
253
  entity_type = value["TYPE"]
254
- if entity_type in entity_json:
255
  anonymization_type = entity_json[entity_type]
256
  if anonymization_type == "RANDOM":
257
  entities.append([key, value["RANDOM"]])
 
251
  entities = []
252
  for key, value in prediction_json.items():
253
  entity_type = value["TYPE"]
254
+ if entity_type.upper() in entity_json:
255
  anonymization_type = entity_json[entity_type]
256
  if anonymization_type == "RANDOM":
257
  entities.append([key, value["RANDOM"]])