Fix output type of multiquantile

ibis-project · Aug 9, 2017 · 05ab434 · 05ab434
1 parent d9a14de
commit 05ab434
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ibis/expr/operations.py b/ibis/expr/operations.py
@@ -727,6 +727,7 @@ class Quantile(Reduction):
                        'midpoint', 'nearest'],
                       name='interpolation',
                       default='linear')]
+
     output_type = rules.scalar_output(_coerce_integer_to_double_type)
 
 
@@ -740,6 +741,9 @@ class MultiQuantile(Quantile):
                       name='interpolation',
                       default='linear')]
 
+    def output_type(self):
+        return dt.Array(_coerce_integer_to_double_type(self)).scalar_type()
+
 
 class VarianceBase(Reduction):