Fix issues selecting and values function on cuDFInterface
@@ -142,8 +142,10 @@ class cuDFInterface(PandasInterface):
return data
elif compute:
return data.to_array()
+ try:
return data.values
-
+ except Exception:
+ return data.values_host
@classmethod
def groupby(cls, dataset, dimensions, container_type, group_type, **kwargs):
@@ -194,7 +196,7 @@ class cuDFInterface(PandasInterface):
for dim, sel in selection.items():
if isinstance(sel, tuple):
sel = slice(*sel)
- arr = cls.values(dataset, dim, compute=False)
+ arr = cls.values(dataset, dim, keep_index=True)
if util.isdatetime(arr) and util.pd:
try:
sel = util.parse_datetime_selection(sel)
