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)
if isinstance(cf, np.ndarray):
cf = tuple(cf.tolist())
cf = cf.tolist()
assert isinstance(cf, list), (
f"Constraint features must be a list. "
f"Found {type(cf).__name__} instead for cname={cname}."

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

Loading…
Cancel
Save