Skip to content

Commit

Permalink
make @scalar_rule use ::Tuple
Browse files Browse the repository at this point in the history
  • Loading branch information
MasonProtter authored Jan 15, 2024
1 parent 77748b3 commit 82d6dc9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/rule_definition_tools.jl
Original file line number Diff line number Diff line change
Expand Up @@ -209,7 +209,7 @@ function scalar_frule_expr(__source__, f, call, setup_stmts, inputs, partials)
return @strip_linenos quote
# _ is the input derivative w.r.t. function internals. since we do not
# allow closures/functors with @scalar_rule, it is always ignored
function ChainRulesCore.frule((_, $(Δs...)), ::Core.Typeof($f), $(inputs...))
function ChainRulesCore.frule((_, $(Δs...))::Tuple, ::Core.Typeof($f), $(inputs...))
$(__source__)
$(esc()) = $call
$(setup_stmts...)
Expand Down

0 comments on commit 82d6dc9

Please sign in to comment.