Make get_instance_features return np.ndarray

2025-12-06 09:28:51 -06:00 · 2021-08-09 14:02:14 -05:00
parent 47d3011808
commit 56b39b6c9c
6 changed files with 29 additions and 16 deletions
--- a/miplearn/solvers/pyomo/base.py
+++ b/miplearn/solvers/pyomo/base.py
@@ -622,11 +622,13 @@ class PyomoTestInstanceKnapsack(Instance):
        return model

    @overrides
-    def get_instance_features(self) -> List[float]:
-        return [
-            self.capacity,
-            np.average(self.weights),
-        ]
+    def get_instance_features(self) -> np.ndarray:
+        return np.array(
+            [
+                self.capacity,
+                np.average(self.weights),
+            ]
+        )

    @overrides
    def get_variable_features(self) -> Dict[str, List[float]]: