Constraint features: Fix conversion to list

master
Alinson S. Xavier 4 years ago
parent 659131c8cf
commit fa969cf066
No known key found for this signature in database
GPG Key ID: DCA0DAD4D2F58624

@ -243,7 +243,7 @@ class FeaturesExtractor:
) )
cf = instance.get_constraint_features(cname) cf = instance.get_constraint_features(cname)
if isinstance(cf, np.ndarray): if isinstance(cf, np.ndarray):
cf = tuple(cf.tolist()) cf = cf.tolist()
assert isinstance(cf, list), ( assert isinstance(cf, list), (
f"Constraint features must be a list. " f"Constraint features must be a list. "
f"Found {type(cf).__name__} instead for cname={cname}." f"Found {type(cf).__name__} instead for cname={cname}."

@ -9,7 +9,7 @@ with open("README.md", "r") as fh:
setup( setup(
name="miplearn", name="miplearn",
version="0.2.0.dev7", version="0.2.0.dev8",
author="Alinson S. Xavier", author="Alinson S. Xavier",
author_email="axavier@anl.gov", author_email="axavier@anl.gov",
description="Extensible framework for Learning-Enhanced Mixed-Integer Optimization", description="Extensible framework for Learning-Enhanced Mixed-Integer Optimization",

Loading…
Cancel
Save