From 4a529119246568b7dae70c91b5758fad94f08acd Mon Sep 17 00:00:00 2001
From: "Alinson S. Xavier" <git@axavier.org>
Date: Wed, 4 Aug 2021 13:54:14 -0500
Subject: [PATCH] AlvLouWeh2017: Replace non-finite features by constant

---
 miplearn/features/extractor.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/miplearn/features/extractor.py b/miplearn/features/extractor.py
index 74c359d..6040527 100644
--- a/miplearn/features/extractor.py
+++ b/miplearn/features/extractor.py
@@ -320,8 +320,10 @@ class FeaturesExtractor:
                 else:
                     f.append(0.0)
 
-            for v in f:
-                assert isfinite(v), f"non-finite elements detected: {f}"
+            for (i, v) in enumerate(f):
+                if not isfinite(v):
+                    f[i] = 0.0
+
             features.append(f)
         return features