time_binnings: store thresholds as np.array

biolab · Mar 9, 2022 · c00df78 · c00df78
1 parent f07d597
commit c00df78
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/Orange/preprocess/discretize.py b/Orange/preprocess/discretize.py
@@ -386,7 +386,7 @@ def _time_binnings(mn, mx, min_pts, max_pts):
         if not times:
             continue
         times = [time.struct_time(t + (0, 0, 0)) for t in times]
-        thresholds = [calendar.timegm(t) for t in times]
+        thresholds = np.array([calendar.timegm(t) for t in times])
         labels = [time.strftime(fmt, t) for t in times]
         short_labels = _simplified_labels(labels)
         if place == 2 and step >= 7:

diff --git a/Orange/preprocess/tests/test_discretize.py b/Orange/preprocess/tests/test_discretize.py
@@ -39,7 +39,8 @@ def tr(ss):
 
         def testbin(start, end):
             bins = _time_binnings(create(*start), create(*end), 3, 51)
-            return [(bin.width_label, tr(bin.short_labels), bin.thresholds)
+            return [(bin.width_label, tr(bin.short_labels),
+                     list(bin.thresholds))
                     for bin in reversed(bins)]
 
         self.assertEqual(