Update test/regression/low_precision.jl

JuliaAI · Oct 16, 2023 · 37a190e · 37a190e
1 parent 385a066
commit 37a190e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/regression/low_precision.jl b/test/regression/low_precision.jl
@@ -113,6 +113,8 @@
     model = build_forest(labels, features)
     preds = apply_forest(model, features)
     @test typeof(preds) == Vector{Float16}
+    # Verify that the `preds` were calculated based on `labels` of the same type.
+    # If the code at some point converts the numbers to, say, `Float64`, then this test will fail.
     @test !all(x->(x in labels), preds)
 
     preds_MT = apply_forest(model, features; use_multithreading=true)