mirror of
https://github.com/ANL-CEEESA/MIPLearn.git
synced 2025-12-08 10:28:52 -06:00
Make all before/solve callbacks receive same parameters
This commit is contained in:
@@ -86,7 +86,14 @@ def test_lazy_before():
|
||||
component.classifiers["a"].predict_proba = Mock(return_value=[[0.95, 0.05]])
|
||||
component.classifiers["b"].predict_proba = Mock(return_value=[[0.02, 0.80]])
|
||||
|
||||
component.before_solve_mip(solver, instances[0], models[0])
|
||||
component.before_solve_mip(
|
||||
solver=solver,
|
||||
instance=instances[0],
|
||||
model=models[0],
|
||||
stats=None,
|
||||
features=None,
|
||||
training_data=None,
|
||||
)
|
||||
|
||||
# Should ask classifier likelihood of each constraint being violated
|
||||
expected_x_test_a = np.array([[67.0, 21.75, 1287.92]])
|
||||
|
||||
Reference in New Issue
Block a user