Fix bug in PartitionEdge with mixed vertex types, generalize `parti…

…tioned_vertices` to `AbstractGraph`
ITensor · Jan 23, 2024 · 569e829 · 569e829
2 parents ccefa64 + 36de13c
commit 569e829
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/Graphs/partitionedgraphs/partitionedge.jl b/src/Graphs/partitionedgraphs/partitionedge.jl
@@ -1,4 +1,4 @@
-struct PartitionEdge{V,E<:AbstractEdge{<:V}} <: AbstractPartitionEdge{V}
+struct PartitionEdge{V,E<:AbstractEdge{V}} <: AbstractPartitionEdge{V}
   edge::E
 end
 

diff --git a/src/Graphs/partitionedgraphs/partitioning.jl b/src/Graphs/partitionedgraphs/partitioning.jl
@@ -90,7 +90,7 @@ function partitioned_vertices(
 end
 
 function partitioned_vertices(
-  g::AbstractNamedGraph; npartitions=nothing, nvertices_per_partition=nothing, kwargs...
+  g::AbstractGraph; npartitions=nothing, nvertices_per_partition=nothing, kwargs...
 )
   vertex_partitions = partitioned_vertices(
     parent_graph(g); npartitions, nvertices_per_partition, kwargs...