Convert InstanceFeatures into dataclass

2025-12-09 19:08:51 -06:00 · 2021-04-05 20:02:24 -05:00
parent d79eec5da6
commit 94084e0669
7 changed files with 30 additions and 27 deletions
--- a/miplearn/features.py
+++ b/miplearn/features.py
@@ -120,7 +120,7 @@ class FeaturesExtractor:
        for (cid, cdict) in features.constraints.items():
            if cdict["Lazy"]:
                lazy_count += 1
-        return {
-            "User features": user_features,
-            "Lazy constraint count": lazy_count,
-        }
+        return InstanceFeatures(
+            user_features=user_features,
+            lazy_constraint_count=lazy_count,
+        )