update kmeans.jl

Author: joannajzou

src/subsetselection/kmeans.jl

@@ -2,13 +2,13 @@ import PotentialLearning: get_random_subset
 export KMeans
 
 struct KMeans <: SubsetSelector
-    assign :: Vector
-    c :: Vector
+    assign :: Vector{<:Int}
+    c :: Vector{<:Int}
     batch_size :: Int
 end
-function KMeans(assign::Vector, c::Vector, batch_size::Int)
-    return KMeans(assign, c, batch_size)
-end
+# function KMeans(assign::Vector, c::Vector, batch_size::Int)
+#     return KMeans(assign, c, batch_size)
+# end
 function KMeans(desc::Vector, k::Int, batch_size::Int)
     A = Matrix(reduce(hcat, desc))
     res = kmeans(A, k)