ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-07 09:58:51 -06:00
Code Issues Packages Projects Releases Wiki Activity

Only include static features in after-load

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-13 16:08:30 -05:00
parent 8f41278713
commit ef7a50e871
11 changed files with 170 additions and 182 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/components/test_primal.py
View File

@@ -27,6 +27,7 @@ from miplearn.solvers.tests import assert_equals
def sample() -> Sample:
sample = Sample(
after_load=Features(
instance=InstanceFeatures(),
variables={
"x[0]": Variable(category="default"),
"x[1]": Variable(category=None),
@@ -35,7 +36,6 @@ def sample() -> Sample:
},
),
after_lp=Features(
instance=InstanceFeatures(),
variables={
"x[0]": Variable(),
"x[1]": Variable(),
@@ -52,7 +52,7 @@ def sample() -> Sample:
}
),
)
sample.after_lp.instance.to_list = Mock(return_value=[5.0]) # type: ignore
sample.after_load.instance.to_list = Mock(return_value=[5.0]) # type: ignore
sample.after_lp.variables["x[0]"].to_list = Mock( # type: ignore
return_value=[0.0, 0.0]
)