fix diag(nothing) in mve

jbytecode · May 26, 2024 · 7422251 · 7422251
1 parent 4b9e885
commit 7422251
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/mve.jl b/src/mve.jl
@@ -26,9 +26,13 @@ function enlargesubset(initialsubset, data::AbstractMatrix, h::Int)
         covmatrix = cov(data[basicsubset, :])
         md2mat =
             mahalanobisSquaredMatrix(data, meanvector=meanvector, covmatrix=covmatrix)
-        md2 = diag(md2mat)
-        md2sortedindex = sortperm(md2)
-        basicsubset = md2sortedindex[1:(length(basicsubset)+1)]
+        if !isnothing(md2mat)
+            md2 = diag(md2mat)
+            md2sortedindex = sortperm(md2)
+            basicsubset = md2sortedindex[1:(length(basicsubset)+1)]
+        else
+            basicsubset = 1:(length(basicsubset)+1)
+        end 
     end
     return basicsubset
 end