issue when running the example kge-predict-transe-pyg-train.ipynb

[ladwis]

Describe the bug
In the Jupyter notebook of "kge-predict-transe-pyg-train.ipynb", the function create_data_from_graph(relationship_type) doesn't produce the result expected. Instead, an error occurs with the message Index contains duplicate entries, cannot reshape when calling the function train_tensor_data = create_data_from_graph(["TRAIN"]. Plus, "TRAIN" must be between square brackets "[]".

To Reproduce

graphdatascience library version: X.Y
GDS plugin version: X.Y.Z
Python version: X.Y.Z
Neo4j version: X.Y.Z
Operating system: (for example Windows 95/Ubuntu 16.04)

Steps to reproduce the behavior:

• Graph/dataset to reproduce the bug. Alternatively describe as much as possible the data, e.g., graph size and other schema/structure details.
• Python code
• Other queries not run through the python client

Expected behavior

Additional context

[orazve]

Hello,

Did you use the example notebook without any changes? If you have changes, could you please show the minimal working example to reproduce the problem and full error message with stack trace?

Additionally, could you please provide information regarding the versions of the software you are using, including the GDS Python client, GDS, Neo4j, Python, and Pandas versions?

[adamnsch]

I'll also add that we run this notebook daily in as part of our testing, and there it works fine.

[adamnsch]

This issue will be closed due to inactivity. @ladwis please feel free to reopen the issue if need be. Thank you