fix example

benchmarks/bf_bench.jl

@@ -1,5 +1,5 @@
 using SpinGlassNetworks
-using LightGraphs
+using Graphs
 using SpinGlassExhaustive
 using LinearAlgebra
 using CUDA
@@ -13,6 +13,7 @@ function bench_cpu(instance::String, max_states::Int=100)
     ig = ising_graph(instance)
     cl = split_into_clusters(ig, super_square_lattice((m, n, t)))
     @time sp = brute_force(cl[1, 1], num_states=max_states)
+    Spectrum
     sp
 end
 
@@ -36,7 +37,9 @@ sp_gpu = bench_gpu("$(@__DIR__)/pegasus_droplets/2_2_3_00.txt")
 sp_gpu = bench_gpu("$(@__DIR__)/pegasus_droplets/2_2_3_00.txt")
 
 @assert sp_gpu.energies ≈ sp_cpu.energies
-@assert sp_gpu.states == sp_cpu.states
+@show sp_gpu.states
+@show sp_cpu.states[1]
+@assert sp_gpu.states == sp_cpu.states[1]
 
 """
 $(SIGNATURES)

src/ising.jl

@@ -2,7 +2,7 @@ export energy_qubo, energy, kernel, kernel_qubo, kernel_part, exhaustive_search,
 
 struct Spectrum
     energies::AbstractVector
-    states::Vector{Int}
+    states::Union{VecOrMat{Int}, VecOrMat{Int32}}
 end
 
 """

src/naive.jl

@@ -49,5 +49,5 @@ function SpinGlassNetworks.brute_force(
     perm = sortperm(energies)[1:num_states]
     energies_cpu = Array(view(energies, perm))
     σ_cpu = Array(view(σ, :, perm))
-    Spectrum(energies_cpu, [σ_cpu[:, i] for i ∈ 1:size(σ_cpu, 2)])
+    Spectrum(energies_cpu, σ_cpu)
 end