diff --git a/changelog.md b/changelog.md index 54e3a0b..c83f505 100644 --- a/changelog.md +++ b/changelog.md @@ -13,6 +13,8 @@ ## Bug fixes +* Make sure that temporary internal node properties are not included in the visualization output. + ## Improvements diff --git a/python-wrapper/src/neo4j_viz/gds.py b/python-wrapper/src/neo4j_viz/gds.py index 338de75..aff8ad7 100644 --- a/python-wrapper/src/neo4j_viz/gds.py +++ b/python-wrapper/src/neo4j_viz/gds.py @@ -119,6 +119,10 @@ def from_gds( node_properties = [property_name] node_dfs = _fetch_node_dfs(gds, G_fetched, node_properties, G_fetched.node_labels()) + if property_name is not None: + for df in node_dfs.values(): + df.drop(columns=[property_name], inplace=True) + rel_df = _fetch_rel_df(gds, G_fetched) finally: if G_fetched.name() != G.name(): diff --git a/python-wrapper/tests/test_gds.py b/python-wrapper/tests/test_gds.py index fda1caf..773a21d 100644 --- a/python-wrapper/tests/test_gds.py +++ b/python-wrapper/tests/test_gds.py @@ -276,6 +276,9 @@ def test_from_gds_sample(gds: Any) -> None: ): VG = from_gds(gds, G) + # Make sure internal temporary properties are not present + assert set(VG.nodes[0].properties.keys()) == {"labels"} + assert len(VG.nodes) >= 9_500 assert len(VG.nodes) <= 10_500 assert len(VG.relationships) >= 9_500