Merge pull request #197 from PixelgenTechnologies/fix/exe-2058-graph-…

…error-for-missing-component

Fixing the error output for calling pixel graph function with non-existing component name

src/pixelator/pixeldataset/__init__.py

@@ -245,7 +245,7 @@ def graph(
         """
         if component_id:
             potential_component = self.edgelist_lazy.filter(
-                pl.col("component") == component_id
+                pl.col("component").cast(pl.String) == component_id
             )
             if potential_component.head(1).collect().is_empty():
                 raise KeyError(f"{component_id} not found in edgelist")