bug fix in entity resolution viz

JohnSnowLabs · Feb 28, 2022 · fb5403a · fb5403a
1 parent 6676b9f
commit fb5403a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sparknlp_display/entity_resolution.py b/sparknlp_display/entity_resolution.py
@@ -182,6 +182,6 @@ def display(self, result, label_col, resolution_col, document_col='document', ra
                 f_.write(html_content_save)
 
         if return_html:
-            return 
+            return html_content_save
         else:
             return display(HTML(html_content_save))