Fix bug in JsonAnalysisBuilder entity mapping

microsoft · Nov 23, 2023 · 354e223 · 354e223
1 parent 15e03c3
commit 354e223
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/presidio-structured/presidio_structured/analysis_builder.py b/presidio-structured/presidio_structured/analysis_builder.py
@@ -71,9 +71,9 @@ def _generate_analysis_from_results_json(
                     result.recognizer_results, prefix=current_key + "."
                 )
                 mappings.update(nested_mappings.entity_mapping)
-
-            if result.recognizer_results > 0:
-                mappings[current_key] = result.recognizer_results[0].entity_type
+            first_recognizer_result = next(iter(result.recognizer_results), None)
+            if first_recognizer_result is not None:
+                mappings[current_key] = first_recognizer_result.entity_type
         return StructuredAnalysis(entity_mapping=mappings)