EllipticEnvelopeClassifier: Don't check for classes

Author: janezd

Orange/classification/elliptic_envelope.py

@@ -2,6 +2,7 @@
 
 from Orange.base import SklLearner, SklModel
 from Orange.data import Table
+from Orange.preprocess import Continuize, RemoveNaNColumns, SklImpute
 
 __all__ = ["EllipticEnvelopeLearner"]
 
@@ -27,6 +28,7 @@ def mahalanobis(self, observations):
 class EllipticEnvelopeLearner(SklLearner):
     __wraps__ = skl_covariance.EllipticEnvelope
     __returns__ = EllipticEnvelopeClassifier
+    preprocessors = [Continuize(), RemoveNaNColumns(), SklImpute()]
 
     def __init__(self, store_precision=True, assume_centered=False,
                  support_fraction=None, contamination=0.1,

Orange/widgets/data/tests/test_owoutliers.py

@@ -37,3 +37,11 @@ def test_memory_error(self):
             side_effect=MemoryError):
             self.send_signal("Data", data)
             self.assertTrue(self.widget.Error.memory_error.is_shown())
+
+    def test_nans(self):
+        """Widget does not crash with nans"""
+        a = np.arange(20, dtype=float).reshape(4, 5)
+        a[0, 0] = np.nan
+        data = Table(a)
+        self.send_signal(self.widget.Inputs.data, data)
+        self.assertIsNot(self.get_output(self.widget.Outputs.inliers), None)