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

Move user_cuts/lazy_enforced to sample.data

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-07-01 08:37:31 -05:00
parent 80281df8d8
commit 061b1349fe
10 changed files with 56 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/components/test_dynamic_lazy.py
View File

@@ -28,11 +28,11 @@ def training_instances() -> List[Instance]:
samples_0 = [
Sample(
after_load=Features(instance=InstanceFeatures()),
after_mip=Features(extra={"lazy_enforced": {"c1", "c2"}}),
data={"lazy_enforced": {"c1", "c2"}},
),
Sample(
after_load=Features(instance=InstanceFeatures()),
after_mip=Features(extra={"lazy_enforced": {"c2", "c3"}}),
data={"lazy_enforced": {"c2", "c3"}},
),
]
samples_0[0].after_load.instance.to_list = Mock(return_value=[5.0]) # type: ignore
@@ -57,7 +57,7 @@ def training_instances() -> List[Instance]:
samples_1 = [
Sample(
after_load=Features(instance=InstanceFeatures()),
after_mip=Features(extra={"lazy_enforced": {"c3", "c4"}}),
data={"lazy_enforced": {"c3", "c4"}},
)
]
samples_1[0].after_load.instance.to_list = Mock(return_value=[8.0]) # type: ignore