Skip to content

Commit 220bed4

Browse files
committed
fix broadcasting call with aliases
1 parent ab7e603 commit 220bed4

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/model.jl

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -305,10 +305,14 @@ function generate_model_expression(backend, model_options, model_specification)
305305

306306
# Step 1: Inject node's aliases
307307
ms_body = postwalk(ms_body) do expression
308-
if @capture(expression, (lhs_ ~ rhs_ where { options__ }) | (lhs_ .~ rhs_ where { options__ }))
308+
if @capture(expression, lhs_ ~ rhs_ where { options__ })
309309
return :($lhs ~ $(write_inject_tilderhs_aliases(backend, model, rhs)) where { $(options...) })
310-
elseif @capture(expression, (lhs_ ~ rhs_) | (lhs_ .~ rhs_))
310+
elseif @capture(expression, lhs_ .~ rhs_ where { options__ })
311+
return :($lhs .~ $(write_inject_tilderhs_aliases(backend, model, rhs)) where { $(options...) })
312+
elseif @capture(expression, lhs_ ~ rhs_)
311313
return :($lhs ~ $(write_inject_tilderhs_aliases(backend, model, rhs)))
314+
elseif @capture(expression, lhs_ .~ rhs_)
315+
return :($lhs .~ $(write_inject_tilderhs_aliases(backend, model, rhs)))
312316
else
313317
return expression
314318
end

0 commit comments

Comments
 (0)