Skip to content

Commit 5d48d24

Browse files
committed
Modify import MarabouCore to MarabouUtils for Equation
1 parent 7991d8c commit 5d48d24

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

airobas/blocks_hub/marabou_block.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
# from tensorflow.keras.models import Sequential
1616
from keras.layers import Activation, Dense
1717
from keras.models import Sequential
18-
from maraboupy import Marabou, MarabouCore
18+
from maraboupy import Marabou, MarabouCore, MarabouUtils
1919
from maraboupy.MarabouNetwork import MarabouNetwork # (pip install maraboupy)
2020

2121
logger = logging.getLogger(__name__)
@@ -187,11 +187,11 @@ def solve_stability_property(network: MarabouSequential, x_min, x_max, y_min, y_
187187
for i in range(network.get_output_dim()):
188188
if np.isinf(y_min[i]) or np.isinf(y_max[i]):
189189
continue
190-
equ_l = MarabouCore.Equation(MarabouCore.Equation.LE) # greater or equal >= scalar
190+
equ_l = MarabouUtils.Equation(MarabouCore.Equation.LE) # greater or equal >= scalar
191191
equ_l.addAddend(1, network.outputVars[0][0][i])
192192
equ_l.setScalar(y_min[i])
193193
# equ_l : f(x)[i]< Y_min[i]
194-
equ_u = MarabouCore.Equation(MarabouCore.Equation.GE) # greater or equal >= scalar
194+
equ_u = MarabouUtils.Equation(MarabouCore.Equation.GE) # greater or equal >= scalar
195195
equ_u.addAddend(1, network.outputVars[0][0][i])
196196
equ_u.setScalar(y_max[i])
197197
# equ_u : f(x)[i]> Y_max[i]

0 commit comments

Comments
 (0)