Directional Relations supported

JohnSnowLabs · Nov 15, 2022 · d2f6483 · d2f6483
1 parent 47b6d36
commit d2f6483
Show file tree

Hide file tree

Showing 8 changed files with 7 additions and 7 deletions.
diff --git a/build/lib/sparknlp_display/VERSION b/build/lib/sparknlp_display/VERSION
@@ -1 +1 @@
-4.1
+4.2
diff --git a/build/lib/sparknlp_display/relation_extraction.py b/build/lib/sparknlp_display/relation_extraction.py
@@ -356,8 +356,8 @@ def __gen_graph(self, rdf, selected_text, exclude_relations, show_relations):
             if row.result.lower().strip() not in self.color_dict:
                 self.color_dict[row.result.lower().strip()] = self.__get_color(row.result.lower().strip())
 
-            d_key2 = all_done[int(row.metadata['entity2_begin'])]
-            d_key1 = all_done[int(row.metadata['entity1_begin'])]
+            d_key2 = all_done[int(row.metadata['entity1_begin'])]
+            d_key1 = all_done[int(row.metadata['entity2_begin'])]
             this_dist = abs(d_key2[0] - d_key1[0]) + abs (d_key2[1]-d_key1[1])
             relation_distances.append(this_dist)
             relation_coordinates.append((d_key2, d_key1, row.result))

diff --git a/dist/spark-nlp-display-4.1.tar.gz b/dist/spark-nlp-display-4.1.tar.gz
diff --git a/dist/spark-nlp-display-4.2.tar.gz b/dist/spark-nlp-display-4.2.tar.gz
diff --git a/dist/spark_nlp_display-4.1-py3-none-any.whl → dist/spark_nlp_display-4.2-py3-none-any.whl b/dist/spark_nlp_display-4.1-py3-none-any.whl → dist/spark_nlp_display-4.2-py3-none-any.whl
diff --git a/spark_nlp_display.egg-info/PKG-INFO b/spark_nlp_display.egg-info/PKG-INFO
@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: spark-nlp-display
-Version: 4.1
+Version: 4.2
 Summary: Visualization package for Spark NLP
 Home-page: http://nlp.johnsnowlabs.com
 Author: John Snow Labs

diff --git a/sparknlp_display/VERSION b/sparknlp_display/VERSION
@@ -1 +1 @@
-4.1
+4.2
diff --git a/sparknlp_display/relation_extraction.py b/sparknlp_display/relation_extraction.py
@@ -356,8 +356,8 @@ def __gen_graph(self, rdf, selected_text, exclude_relations, show_relations):
             if row.result.lower().strip() not in self.color_dict:
                 self.color_dict[row.result.lower().strip()] = self.__get_color(row.result.lower().strip())
 
-            d_key2 = all_done[int(row.metadata['entity2_begin'])]
-            d_key1 = all_done[int(row.metadata['entity1_begin'])]
+            d_key2 = all_done[int(row.metadata['entity1_begin'])]
+            d_key1 = all_done[int(row.metadata['entity2_begin'])]
             this_dist = abs(d_key2[0] - d_key1[0]) + abs (d_key2[1]-d_key1[1])
             relation_distances.append(this_dist)
             relation_coordinates.append((d_key2, d_key1, row.result))