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

Convert ConstraintFeatures to dataclass

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-05 20:12:07 -05:00
parent 94084e0669
commit aeed338837
6 changed files with 64 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
tests/components/test_lazy_static.py
View File

@@ -17,6 +17,7 @@ from miplearn.types import (
Features,
LearningSolveStats,
InstanceFeatures,
ConstraintFeatures,
)
@@ -35,31 +36,31 @@ def features() -> Features:
lazy_constraint_count=4,
),
constraints={
"c1": {
"Category": "type-a",
"User features": [1.0, 1.0],
"Lazy": True,
},
"c2": {
"Category": "type-a",
"User features": [1.0, 2.0],
"Lazy": True,
},
"c3": {
"Category": "type-a",
"User features": [1.0, 3.0],
"Lazy": True,
},
"c4": {
"Category": "type-b",
"User features": [1.0, 4.0, 0.0],
"Lazy": True,
},
"c5": {
"Category": "type-b",
"User features": [1.0, 5.0, 0.0],
"Lazy": False,
},
"c1": ConstraintFeatures(
category="type-a",
user_features=[1.0, 1.0],
lazy=True,
),
"c2": ConstraintFeatures(
category="type-a",
user_features=[1.0, 2.0],
lazy=True,
),
"c3": ConstraintFeatures(
category="type-a",
user_features=[1.0, 3.0],
lazy=True,
),
"c4": ConstraintFeatures(
category="type-b",
user_features=[1.0, 4.0, 0.0],
lazy=True,
),
"c5": ConstraintFeatures(
category="type-b",
user_features=[1.0, 5.0, 0.0],
lazy=False,
),
},
)