Merge pull request #188 from neo4j/drop-temp-internal-props

Make sure to drop temp internal node props in `from_gds`

Author: FlorentinD

changelog.md

@@ -13,6 +13,8 @@
 
 ## Bug fixes
 
+* Make sure that temporary internal node properties are not included in the visualization output.
+
 
 ## Improvements
 

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():

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