renard-pipeline 0.6.1__py3-none-any.whl → 0.6.3__py3-none-any.whl

renard/graph_utils.py

@@ -124,6 +124,15 @@ def dynamic_graph_to_gephi_graph(graphs: List[nx.Graph]) -> nx.Graph:
     G = nx.Graph()
 
     for H_i, H in enumerate(graphs):
+
+        # update node attributes
+        for u in H.nodes:
+            G.add_node(u)
+            for key, value in H.nodes[u].items():
+                # format : (VALUE, START, END)
+                series = G.nodes[u].get(key, [])
+                G.nodes[u][key] = series + [(value, H_i, H_i + 1)]
+
         for u, v in H.edges:
             # create a new edge between two characters if it doesn't
             # exist yet

renard/pipeline/graph_extraction.py

@@ -306,8 +306,11 @@ class CoOccurrencesGraphExtractor(PipelineStep):
 
         # * Construct graph from co-occurence matrix
         G = nx.Graph()
-        for character, _ in mentions:
-            G.add_node(character)
+        for character, mention in mentions:
+            # NOTE: we add an 'entity_type' attribute. This is useful
+            # when using the 'additional_ner_classes' option, to
+            # differentiate between different entity types.
+            G.add_node(character, entity_type=mention.tag)
 
         for i, (char1, mention1) in enumerate(mentions):
             for j, (char2, mention2) in enumerate(mentions):

renard/pipeline/stanford_corenlp.py

@@ -72,7 +72,7 @@ class StanfordCoreNLPPipeline(PipelineStep):
     .. note::
 
         The Stanford CoreNLP pipeline requires the ``stanza`` library.
-        You can install it with poetry using ``poetry install -E stanza``.
+        You can install it with poetry using ``pip install stanza``.
 
     .. warning::
 

{renard_pipeline-0.6.1.dist-info → renard_pipeline-0.6.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: renard-pipeline
-Version: 0.6.1
+Version: 0.6.3
 Summary: Relationships Extraction from NARrative Documents
 Home-page: https://github.com/CompNet/Renard
 License: GPL-3.0-only
@@ -13,9 +13,6 @@ Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
-Provides-Extra: spacy
-Provides-Extra: stanza
-Requires-Dist: coreferee (>=1.4,<2.0) ; extra == "spacy"
 Requires-Dist: datasets (>=3.0,<4.0)
 Requires-Dist: grimbert (>=0.1,<0.2)
 Requires-Dist: matplotlib (>=3.5,<4.0)
@@ -26,13 +23,10 @@ Requires-Dist: nltk (>=3.9,<4.0)
 Requires-Dist: pandas (>=2.0,<3.0)
 Requires-Dist: pytest (>=8.3.0,<9.0.0)
 Requires-Dist: rank-bm25 (>=0.2.2,<0.3.0)
-Requires-Dist: spacy (>=3.5,<4.0) ; extra == "spacy"
-Requires-Dist: spacy-transformers (>=1.3,<2.0) ; extra == "spacy"
-Requires-Dist: stanza (>=1.3,<2.0) ; extra == "stanza"
 Requires-Dist: tibert (>=0.5,<0.6)
 Requires-Dist: torch (>=2.0.0,!=2.0.1)
 Requires-Dist: tqdm (>=4.62.3,<5.0.0)
-Requires-Dist: transformers (>=4.36,<5.0)
+Requires-Dist: transformers (>=4.37,<5.0)
 Project-URL: Documentation, https://compnet.github.io/Renard/
 Project-URL: Repository, https://github.com/CompNet/Renard
 Description-Content-Type: text/markdown

{renard_pipeline-0.6.1.dist-info → renard_pipeline-0.6.3.dist-info}/RECORD

@@ -1,5 +1,5 @@
 renard/gender.py,sha256=HDtJQKOqIkV8F-Mxva95XFXWJoKRKckQ3fc93OBM6sw,102
-renard/graph_utils.py,sha256=EV0_56KtI3VOElCu7wxd2kL8QVPsOu7itE6wGJAJsNA,6073
+renard/graph_utils.py,sha256=fcxXXcpzOkUJNgps0zgdoWJNydwkWwJJLJ6OlViNxTA,6373
 renard/ner_utils.py,sha256=SFZoyJM6c2avE7-NDkCSzkx-O8ppzS00a8EyHt64iGI,11628
 renard/nltk_utils.py,sha256=mUJiwMrEDZV4Fla7WuMR-hA_OC2ZIwSXgW_0Ew18VSo,977
 renard/pipeline/__init__.py,sha256=8Yim2mmny8YGvM7N5-na5zK-C9UDxUb77K9ml-VirUA,35
@@ -8,7 +8,7 @@ renard/pipeline/characters_extraction.py,sha256=bMic8dtlYKUmAlTzQqDPraYy5VsGWoGk
 renard/pipeline/core.py,sha256=LILUIQZp9f3FzqjBocUS7dKzX7lHQQVdL29jyqU1UeY,27754
 renard/pipeline/corefs/__init__.py,sha256=9c9AaXBcRrDBf1jhTtJ7DyjOJhX_Zej3FjlcGak7MK8,44
 renard/pipeline/corefs/corefs.py,sha256=d47Sd8ekwhQQV6rQ0F9QyAX2GOTqUnkDUA-eKgMtMS4,11417
-renard/pipeline/graph_extraction.py,sha256=Ga3wfUW9tDtatcTv2taLrNky9jz2wUwZ8uzoXJoSVk8,22928
+renard/pipeline/graph_extraction.py,sha256=CsQTdTKj2JmePgiw37_yF0SHcGTRmp62-8Faouaoty4,23154
 renard/pipeline/ner/__init__.py,sha256=Dqxcf_EKhK1UwiCscZ3gGHInlcxJyvpR4o-ZCLEyV48,38
 renard/pipeline/ner/ner.py,sha256=8zUtaqaGNirfGFRyMpDzdqtO3abrRLyLtjmwnqBNwUI,9893
 renard/pipeline/ner/retrieval.py,sha256=JIU3fi0Q1gl_YGP6kYx6zC9xz4UN6gnqdVuzWVXzzyM,12853
@@ -18,7 +18,7 @@ renard/pipeline/progress.py,sha256=PJ174ssaqr5qHaTrVQ8HqJtvpvX6QhtHM5PHT893_Xk,2
 renard/pipeline/quote_detection.py,sha256=FyldJhynIT843fB7rwVtHmDZJqTKkjGml6qTLjsIhMA,2045
 renard/pipeline/sentiment_analysis.py,sha256=76MPin4L1-vSswJe5yGrbCSSDim1LYxSEgNj_BdQDvk,1464
 renard/pipeline/speaker_attribution.py,sha256=Uts6JdUo_sbWyIb2AJ6SO5JuUbgROIpcbUNTg4dHo4U,4329
-renard/pipeline/stanford_corenlp.py,sha256=14b6Ee6oPz1EL-bNRT688aNxVTk_Jwa_vJ20FiBODC4,8189
+renard/pipeline/stanford_corenlp.py,sha256=xfauQk5aPLz0xdRbQVeD9x-sL8uqEzxPRhqJtqvDVGg,8183
 renard/pipeline/tokenization.py,sha256=gZP0ZpAa0rhtUDPk6W0PiXRxmiC3IcSyRF_E7KaP19A,2957
 renard/plot_utils.py,sha256=qsQI-wbk_5KCXDvt1tPerq4UW4VWLrJpoCet4qkONwE,3344
 renard/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -33,7 +33,7 @@ renard/resources/pronouns/pronouns.py,sha256=YJ8hM6H8QHrF2Xx6O5blqc-Sqe1D1YFL0sR
 renard/resources/titles/__init__.py,sha256=Jcg4B7stsWiAaXbFgNl_L3ICtCQmFe9bo3YjdkVL50w,45
 renard/resources/titles/titles.py,sha256=GsFccVJuTkgDWiAqWZpFd2R9pGvFKQZBOk4RWWuWDkw,968
 renard/utils.py,sha256=WL6djr3iu5Kzo2Jq6qDllHXgvZcEnmqBxPkQf1drq7c,4072
-renard_pipeline-0.6.1.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-renard_pipeline-0.6.1.dist-info/METADATA,sha256=vijGA3DMBq0Tkn2SJxMKacOw8zI5Z4IDSmIBWBuMEuM,4374
-renard_pipeline-0.6.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-renard_pipeline-0.6.1.dist-info/RECORD,,
+renard_pipeline-0.6.3.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+renard_pipeline-0.6.3.dist-info/METADATA,sha256=70oB4i5Pe74ZUBMTv84qt30o5EcpqtYUYotBUeS2Wyw,4098
+renard_pipeline-0.6.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+renard_pipeline-0.6.3.dist-info/RECORD,,