Skip to content

Commit bf65928

Browse files
committed
Relax dist DSL typechecks to support TrackedReal.
1 parent 924d6cb commit bf65928

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/modeling_library/dist_dsl/dist_dsl.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@ eval_arg(x::TransformedArg, args) =
3333

3434
# Type of SimpleArg must match arg, otherwise a MethodError will be thrown
3535
typecheck_arg(x::SimpleArg{T}, arg::T) where {T} = arg
36+
typecheck_arg(x::SimpleArg{T}, arg::ReverseDiff.TrackedReal{T}) where {T <: Real} = arg
3637

3738
# DistWithArgs
3839
struct DistWithArgs{T}

0 commit comments

Comments
 (0)