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

Primal: Refactoring

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-03-31 09:08:01 -05:00
parent 4f46866921
commit 0f5a6745a4
4 changed files with 32 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
miplearn/features.py
View File

@@ -6,7 +6,7 @@ import numbers
import collections
from typing import TYPE_CHECKING, Dict
from miplearn.types import ModelFeatures, ConstraintFeatures
from miplearn.types import Features, ConstraintFeatures
if TYPE_CHECKING:
from miplearn import InternalSolver, Instance
@@ -19,7 +19,7 @@ class FeaturesExtractor:
) -> None:
self.solver = internal_solver
def extract(self, instance: "Instance") -> ModelFeatures:
def extract(self, instance: "Instance") -> Features:
return {
"Constraints": self._extract_constraints(instance),
"Variables": self._extract_variables(instance),