Testing: Keep 2d array of models when splitting Results by models

Author: janezd

Orange/evaluation/testing.py

@@ -397,7 +397,7 @@ def split_by_model(self):
                 res.probabilities = self.probabilities[(i,), :, :]
 
             if self.models is not None:
-                res.models = self.models[:, i]
+                res.models = self.models[:, i:i+1]
 
             res.failed = [self.failed[i]]
             yield res