From 2ed473d6af3d47536a136585d4b580c9fcbe61c6 Mon Sep 17 00:00:00 2001
From: "Guus@XPS-ubuntu" <guusavis@hotmail.com>
Date: Mon, 5 Aug 2024 15:50:08 -0400
Subject: [PATCH] Type checking for propagate kwargs.

---
 src/propagate.jl | 4 ++--
 test/triples.jl  | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/propagate.jl b/src/propagate.jl
index 4b533ee7..6d5ee5b8 100644
--- a/src/propagate.jl
+++ b/src/propagate.jl
@@ -92,8 +92,8 @@ StochasticTriple of Int64:
 """
 function propagate(f,
         args...;
-        keep_deltas = Val(false),
-        keep_triples = Val(false),
+        keep_deltas::Val = Val(false),
+        keep_triples::Val = Val(false),
         provided_st_rep = nothing,
         deriv = nothing)
     # TODO: support kwargs to f (or just use kwfunc in macro)
diff --git a/test/triples.jl b/test/triples.jl
index 79b8ac28..08d46e6e 100644
--- a/test/triples.jl
+++ b/test/triples.jl
@@ -529,7 +529,7 @@ end
                 out_δ_expected = StochasticAD.structural_map(zero, out)
             end
             input_sts = StochasticAD.structural_map(_form_triple, primals, δs, Δs)
-            out_st = StochasticAD.propagate(f, input_sts...; keep_deltas = Val{test_deltas})
+            out_st = StochasticAD.propagate(f, input_sts...; keep_deltas = Val(test_deltas))
             # Test type
             StochasticAD.structural_map(out_st, out, out_δ_expected,
                 out_Δ_expected) do x_st, x, δ, Δ