neo4j-viz 0.7.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

neo4j_viz/pandas.py

@@ -69,6 +69,9 @@ def _parse_nodes(node_dfs: DFS_TYPE, dropna: bool = False) -> list[Node]:
             mandatory_fields = {}
             properties = {}
             for key, value in row.to_dict().items():
+                if not isinstance(key, str):
+                    key = str(key)
+
                 if key in basic_node_fields_aliases:
                     mandatory_fields[key] = value
                 else:
@@ -98,6 +101,9 @@ def _parse_relationships(rel_dfs: DFS_TYPE, dropna: bool = False) -> list[Relati
             mandatory_fields = {}
             properties = {}
             for key, value in row.to_dict().items():
+                if not isinstance(key, str):
+                    key = str(key)
+
                 if key in basic_rel_field_aliases:
                     mandatory_fields[key] = value
                 else:

neo4j_viz/relationship.py

@@ -64,6 +64,8 @@ class Relationship(
     caption_size: Optional[Union[int, float]] = Field(None, gt=0.0, description="The size of the caption text")
     #: The color of the relationship. Allowed input is for example "#FF0000", "red" or (255, 0, 0)
     color: Optional[ColorType] = Field(None, description="The color of the relationship")
+    # The width of the relationship
+    width: Optional[int | float] = Field(None, gt=0.0, description="The width of the relationship")
     #: Additional properties of the relationship that do not directly impact the visualization
     properties: dict[str, Any] = Field(
         default_factory=dict,