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