fix failure

itholic · Nov 6, 2019 · 37902f9 · 37902f9
1 parent cd598f2
commit 37902f9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/databricks/koalas/frame.py b/databricks/koalas/frame.py
@@ -7540,10 +7540,10 @@ def quantile(self, q=0.5, axis=0, numeric_only=True, accuracy=10000):
         # +---------+---------+
 
         cols_dict = OrderedDict()
-        for column in self.columns:
+        for column in self._internal.data_columns:
             cols_dict[column] = list()
             for i in range(len(quantiles)):
-                cols_dict[column].append(self._internal.scol_for(column).getItem(i).alias(column))
+                cols_dict[column].append(F.col(column).getItem(i).alias(column))
 
         internal_index_column = SPARK_INDEX_NAME_FORMAT(0)
         cols = []