ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-09 02:48: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

22
miplearn/types.py
View File

@@ -81,7 +81,7 @@ LearningSolveStats = TypedDict(
@dataclass
class InstanceFeatures:
user_features: List[float]
user_features: Optional[List[float]] = None
lazy_constraint_count: int = 0
@@ -91,18 +91,14 @@ class VariableFeatures:
user_features: Optional[List[float]] = None
ConstraintFeatures = TypedDict(
"ConstraintFeatures",
{
"RHS": float,
"LHS": Dict[str, float],
"Sense": str,
"Category": Optional[Hashable],
"User features": Optional[List[float]],
"Lazy": bool,
},
total=False,
)
@dataclass
class ConstraintFeatures:
rhs: Optional[float] = None
lhs: Optional[Dict[str, float]] = None
sense: Optional[str] = None
category: Optional[Hashable] = None
user_features: Optional[List[float]] = None
lazy: bool = False
@dataclass