Make Sample abstract; create MemorySample

This commit is contained in:
2021-07-14 08:31:01 -05:00
parent 851b8001bb
commit 235c3e55c2
7 changed files with 30 additions and 15 deletions

View File

@@ -10,14 +10,14 @@ from numpy.testing import assert_array_equal
from miplearn.classifiers import Regressor
from miplearn.components.objective import ObjectiveValueComponent
from miplearn.features.sample import Sample
from miplearn.features.sample import Sample, MemorySample
from miplearn.solvers.learning import LearningSolver
from miplearn.solvers.pyomo.gurobi import GurobiPyomoSolver
@pytest.fixture
def sample() -> Sample:
sample = Sample(
sample = MemorySample(
{
"mip_lower_bound": 1.0,
"mip_upper_bound": 2.0,