neo4j-viz 0.2.5__py3-none-any.whl → 0.2.6__py3-none-any.whl

neo4j_viz/visualization_graph.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 import warnings
 from collections.abc import Iterable
-from typing import Optional
+from typing import Any, Hashable, Optional
 
 from IPython.display import HTML
 from pydantic_extra_types.color import Color, ColorType
@@ -201,7 +201,8 @@ class VisualizationGraph:
         Parameters
         ----------
         property:
-            The property of the nodes to use for coloring.
+            The property of the nodes to use for coloring. The type of this property must be hashable, or be a
+            list, set or dict containing only hashable types.
         colors:
             The colors to use for the nodes. If a dictionary is given, it should map from property to color.
             If an iterable is given, the colors are used in order.
@@ -238,7 +239,11 @@ class VisualizationGraph:
         prop_to_color = {}
         colors_iter = iter(colors)
         for node in self.nodes:
-            prop = getattr(node, property)
+            raw_prop = getattr(node, property)
+            try:
+                prop = self._make_hashable(raw_prop)
+            except ValueError:
+                raise ValueError(f"Unable to color nodes by unhashable property type '{type(raw_prop)}'")
 
             if prop not in prop_to_color:
                 next_color = next(colors_iter, None)
@@ -263,3 +268,22 @@ class VisualizationGraph:
                 f"Ran out of colors for property '{property}'. {len(prop_to_color)} colors were needed, but only "
                 f"{len(set(prop_to_color.values()))} were given, so reused colors"
             )
+
+    @staticmethod
+    def _make_hashable(raw_prop: Any) -> Hashable:
+        prop = raw_prop
+        if isinstance(raw_prop, list):
+            prop = tuple(raw_prop)
+        elif isinstance(raw_prop, set):
+            prop = frozenset(raw_prop)
+        elif isinstance(raw_prop, dict):
+            prop = tuple(sorted(raw_prop.items()))
+
+        try:
+            hash(prop)
+        except TypeError:
+            raise ValueError(f"Unable to convert property '{raw_prop}' of type {type(raw_prop)} to a hashable type")
+
+        assert isinstance(prop, Hashable)
+
+        return prop

{neo4j_viz-0.2.5.dist-info → neo4j_viz-0.2.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: neo4j-viz
-Version: 0.2.5
+Version: 0.2.6
 Summary: A simple graph visualization tool
 Author-email: Neo4j <team-gds@neo4j.org>
 Project-URL: Homepage, https://neo4j.com/

{neo4j_viz-0.2.5.dist-info → neo4j_viz-0.2.6.dist-info}/RECORD

@@ -9,9 +9,9 @@ neo4j_viz/options.py,sha256=QBeHLqHOJFtSCjzHvv-2A9TzTbnIaC6_onQq9fUlT_4,3079
 neo4j_viz/pandas.py,sha256=180SEJGxT16OnXhd2KRisW2rHbrvyJAFZ6MNlhqfKco,1898
 neo4j_viz/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 neo4j_viz/relationship.py,sha256=3Nz49UH3Ae6eTT9Lj6_eqO7nNANpuMGnGfSCL0DABHA,2751
-neo4j_viz/visualization_graph.py,sha256=Y7HrW_E1csn2SbdKjNFvA7lZsJ5qSU29O0DFNOzi4lQ,9009
+neo4j_viz/visualization_graph.py,sha256=ys1gV3Me_1yFAqNknDqWTVZlX26KS_x2IF0dq9hy22Y,9932
 neo4j_viz/resources/nvl_entrypoint/base.js,sha256=FPvDNl_DqFW72SqTrnY5NqIxfs6p66eqFoZJKoY6848,1800059
-neo4j_viz-0.2.5.dist-info/METADATA,sha256=NMDZC6cE-0QVjDtHqRz2aopaw_ep7dBjU-E7EfHgbJM,7003
-neo4j_viz-0.2.5.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-neo4j_viz-0.2.5.dist-info/top_level.txt,sha256=jPUM3z8MOtxqDanc2VzqkxG4HJn8aaq4S7rnCFNk_Vs,10
-neo4j_viz-0.2.5.dist-info/RECORD,,
+neo4j_viz-0.2.6.dist-info/METADATA,sha256=_-gENXULw6rLPwEJEmt3MqYjgrmj_epNJsJPjsG-dOk,7003
+neo4j_viz-0.2.6.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+neo4j_viz-0.2.6.dist-info/top_level.txt,sha256=jPUM3z8MOtxqDanc2VzqkxG4HJn8aaq4S7rnCFNk_Vs,10
+neo4j_viz-0.2.6.dist-info/RECORD,,