Skip to content

Commit 835fec2

Browse files
committed
Small correction to Num handling
1 parent 6564654 commit 835fec2

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

src/relaxation/rules.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,10 +31,11 @@ function transform_rule(::McCormickTransform, ::typeof(exp), yL, yU, ycv, ycc, x
3131
mcv = mid_expr(xcv, xcc, xL)
3232
mcc = mid_expr(xcv, xcc, xU)
3333
rcv = Equation(ycv, exp(mcv))
34-
rcc = Equation(ycc, line_expr(mcc, xL, xU, yL, yU))
34+
rcc = Equation(ycc, line_expr(mcc, xL, xU, exp(xL), exp(xU)))
3535
return rcv, rcc
3636
end
3737

38+
3839
#=
3940
Binary Rules
4041
=#

src/transform/utilities.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -235,7 +235,7 @@ julia> pull_vars(func)
235235
function pull_vars(term::Num)
236236
vars = Num[]
237237
strings = String[]
238-
vars, strings = _pull_vars(term, vars, strings)
238+
vars, strings = _pull_vars(term.val, vars, strings)
239239
vars = vars[sortperm(strings)]
240240
return vars
241241
end
@@ -244,7 +244,7 @@ function pull_vars(terms::Vector{Num})
244244
vars = Num[]
245245
strings = String[]
246246
for term in terms
247-
vars, strings = _pull_vars(term, vars, strings)
247+
vars, strings = _pull_vars(term.val, vars, strings)
248248
end
249249
vars = vars[sortperm(strings)]
250250
return vars

0 commit comments

Comments
 (0)