Merge pull request #131 from invenia/bs/fixes-test-compatibility

Fixes compatibility for tests running against Juliav1.8
invenia · Oct 19, 2022 · 7d643ff · 7d643ff
2 parents 9230661 + 044dc4c
commit 7d643ff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/imputors/knn.jl b/test/imputors/knn.jl
@@ -66,8 +66,8 @@
             knn_imputed = impute(copy(X), Impute.KNN(; k=4); dims=:cols)
             mean_imputed = impute(copy(X), Substitute(); dims=:cols)
 
-            knn_nrmsd = ((i - 1) * knn_nrmsd + nrmsd(data', knn_imputed)) / i
-            mean_nrmsd = ((i - 1) * mean_nrmsd + nrmsd(data', mean_imputed)) / i
+            knn_nrmsd = ((i - 1) * knn_nrmsd + nrmsd(knn_imputed, data')) / i
+            mean_nrmsd = ((i - 1) * mean_nrmsd + nrmsd(mean_imputed, data')) / i
         end
 
         # @show knn_nrmsd mean_nrmsd