# Buggy code
def ensure_not_chunked(arrow_array):
    if isinstance(arrow_array, pa.ChunkedArray):
        if len(arrow_array.chunks) == 0:
            return arrow_array.chunks[0]
        table = pa.Table.from_arrays([arrow_array], ["single"])
        table_concat = table.combine_chunks()
        column = table_concat.columns[0]
        assert column.num_chunks == 1
        arrow_array = column.chunk(0)
    return arrow_array


# Traceback (AttributeError)
AttributeError                            Traceback (most recent call last)
<ipython-input-16-cc5013060237> in <module>
----> 1 df.column_a.mean()

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/expression.py in mean(self, binby, limits, shape, selection, delay, progress)
550         del kwargs['self']
551         kwargs['expression'] = self.expression
--> 552         return self.ds.mean(**kwargs)
553
554     def std(self, binby=[], limits=None, shape=default_shape, selection=False, delay=False, progress=None):

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/dataframe.py in mean(self, expression, binby, limits, shape, selection, delay, progress, edges, array_type)
851         :return: {return_stat_scalar}
852         """
--> 853         return self._compute_agg('mean', expression, binby, limits, shape, selection, delay, edges, progress, array_type=array_type)
854         logger.debug("mean of %r, with binby=%r, limits=%r, shape=%r, selection=%r, delay=%r", expression, binby, limits, shape, selection, delay)
855         expression = _ensure_strings_from_expressions(expression)

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/dataframe.py in _compute_agg(self, name, expression, binby, limits, shape, selection, delay, edges, progress, extra_expressions, array_type)
683             # give errors when evaluated with NaN's present.
684             # TODO: GET RID OF THIS
--> 685             len(self) # fill caches and masks
686             # pass
687         grid = self._create_grid(binby, limits, shape, selection=selection, delay=True)

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/dataframe.py in __len__(self)
3763         else:
3764             if self._cached_filtered_length is None:
-> 3765                self. _cached_filtered_length = int(self.count())
3766             return self._cached_filtered_length
3767

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/dataframe.py in count(self, expression, binby, limits, shape, selection, delay, edges, progress, array_type)
768         :return: {return_stat_scalar}
769         """
--> 770         return self._compute_agg('count', expression, binby, limits, shape, selection, delay, edges, progress, array_type=array_type)
771
772     @delayed

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/dataframe.py in _compute_agg(self, name, expression, binby, limits, shape, selection, delay, edges, progress, extra_expressions, array_type)
742         stats = [compute(expression, grid, selection=selection, edges=edges, progressbar=progressbar) for expression in expressions]
743         var = finish(grid, *stats)
--> 744         return self._delay(delay, var)
745
746     @docsubst

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/dataframe.py in _delay(self, delay, task, progressbar)
1506             return task
1507         else:
-> 1508             self.execute()
1509             return task.get()
1510

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/dataframe.py in execute(self)
343         '''Execute all delayed jobs.'''
344         from .asyncio import just_run
--> 345         just_run(self.execute_async())
346
347     async def execute_async(self):

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/asyncio.py in just_run(coro)
33             nest_asyncio.apply()
34             check_patch_tornado()
---> 35         return loop.run_until_complete(coro)
36     finally:
37         if not had_loop:  # remove loop if we did not have one

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/nest_asyncio.py in run_until_complete(self, future)
96                 raise RuntimeError(
97                     'Event loop stopped before Future completed.')
---> 98             return f.result()
99         finally:
100             self._thread_id = old_thread_id

~/.conda-envs/datalakeutils/lib/python3.6/asyncio/tasks.py in _step(***failed resolving arguments***)
180                 result = coro.send(None)
181             else:
--> 182                 result = coro.throw(exc)
183         except StopIteration as exc:
184             if self._must_cancel:

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/dataframe.py in execute_async(self)
348         '''Async version of execute'''
349         # no need to clear _task_aggs anymore, since they will be removed for the executors' task list
--> 350         await self.executor.execute_async()
351
352     @property

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/execution.py in execute_async(self)
182                                                     all([all(task.signal_progress.emit(p)) for task in tasks]) and
183                                                     all([not task.cancelled for task in tasks]),
--> 184                                                     cancel=lambda: self._cancel(run), unpack=True, run=run):
185                     pass  # just eat all element
186                 logger.debug("executing took %r seconds" % (time.time() - t0))

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/multithreading.py in map_async(self, callable, iterator, count, on_error, progress, cancel, unpack, **kwargs_extra)
88         total = 0
89         for i, value in buffer(enumerate(iterator), self._max_workers + 3):
---> 90             value = await value
91             if value != None:
92                 total += value

~/.conda-envs/datalakeutils/lib/python3.6/asyncio/futures.py in __iter__(self)
325         if not self.done():
326             self._asyncio_future_blocking = True
--> 327             yield self  # This tells Task to wait for completion.
328         assert self.done(), "yield from wasn't used with future"
329         return self.result()  # May raise too.

~/.conda-envs/datalakeutils/lib/python3.6/asyncio/tasks.py in _wakeup(self, future)
248     def _wakeup(self, future):
249         try:
--> 250             future.result()
251         except Exception as exc:
252             # This may also be a cancellation.

~/.conda-envs/datalakeutils/lib/python3.6/asyncio/futures.py in result(self)
241         self._log_traceback = False
242         if self._exception is not None:
--> 243             raise self._exception
244         return self._result
245

~/.conda-envs/datalakeutils/lib/python3.6/concurrent/futures/thread.py in run(self)
54
55         try:
---> 56             result = self.fn(*self.args, **self.kwargs)
57         except BaseException as exc:
58             self.future.set_exception(exc)

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/multithreading.py in <lambda>(value)
84         else:
85             loop = asyncio.get_event_loop()
---> 86             iterator = (loop.run_in_executor(self, lambda value=value: wrapped(value)) for value in iterator)
87
88         total = 0

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/multithreading.py in wrapped(*args, **kwargs)
76                     # print("SLEEP", self._debug_sleep)
77                     time.sleep(self._debug_sleep)
---> 78                 return callable(self.local.index, *args, **kwargs, **kwargs_extra)
79         # convert to list so we can count
80         time_last = time.time() - 100

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/execution.py in process_part(self, thread_index, i1, i2, chunks, run)
232                 assert len(chunk) == N, f'Oops, got a chunk ({name}) of length {len(chunk)} while it is expected to be of length {N} (at {i1}-{i2}'
233             if run.pre_filter:
--> 234                 filter_mask = df.evaluate_selection_mask(None, i1=i1, i2=i2, cache=True)
235                 chunks = {k:vaex.array_types.filter(v, filter_mask) for k, v, in chunks.items()}
236             else:

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/dataframe.py in evaluate_selection_mask(self, name, i1, i2, selection, cache, filtered, pre_filtered)
2578         if name in [None, False] and self.filtered and filtered:
2579             scope_global = scopes._BlockScopeSelection(self, i1, i2, None, cache=cache)
-> 2580             mask_global = scope_global.evaluate(FILTER_SELECTION_NAME)
2581             return vaex.utils.unmask_selection_mask(mask_global)
2582         elif self.filtered and filtered and name != FILTER_SELECTION_NAME:

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/scopes.py in evaluate(self, expression)
181         try:
182             expression = _ensure_string_from_expression(expression)
--> 183             result = eval(expression, expression_namespace, self)
184         except:
185             import traceback as tb

<string> in <module>

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/scopes.py in __getitem__(self, variable)
217                 if variable in self.df.variables:
218                     return self.df.variables[variable]
--> 219                 mask_values = selection.evaluate(self.df, variable, self.i1, self.i2, self.filter_mask)
220
221                 # get a view on a subset of the mask

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/selections.py in evaluate(self, df, name, i1, i2, filter_mask)
128         else:
129             previous_mask = None
--> 130         result = df._evaluate_selection_mask(self.boolean_expression, i1, i2, filter_mask=filter_mask)
131         if isinstance(result, bool):
132             N = i2 - i1

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/dataframe.py in _evaluate_selection_mask(self, name, i1, i2, selection, cache, filter_mask)
2564         i2 = i2 or len(self)
2565         scope = scopes._BlockScopeSelection(self, i1, i2, selection, cache=cache, filter_mask=filter_mask)
-> 2566         mask = scope.evaluate(name)
2567         # TODO: can we do without arrow->numpy conversion?
2568         mask = vaex.array_types.to_numpy(mask)

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/scopes.py in evaluate(self, expression)
181         try:
182             expression = _ensure_string_from_expression(expression)
--> 183             result = eval(expression, expression_namespace, self)
184         except:
185             import traceback as tb

<string> in <module>

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/arrow/numpy_dispatch.py in operator(a, b)
85                 result_data = op['op'](a_data, b_data)
86             if isinstance(a, NumpyDispatch):
---> 87                 result_data = a.add_missing(result_data)
88             if isinstance(b, NumpyDispatch):
89                 result_data = b.add_missing(result_data)

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/arrow/numpy_dispatch.py in add_missing(self, ar)
39             if isinstance(self._array, vaex.array_types.supported_arrow_array_types):
40                 ar = vaex.array_types.to_arrow(ar)
---> 41                 ar = combine_missing(ar, self._array)
42             # else: both numpy, handled by numpy
43         else:

~/.conda-envs/datalakeutils/lib/python3.6/site-packages/vaex/arrow/numpy_dispatch.py in combine_missing(a, b)
11         # not optimal
12         nulls = pc.invert(pc.or_(a.is_null(), b.is_null()))
---> 13         assert nulls.offset == 0
14         nulls_buffer = nulls.buffers()[1]
15         # this is not the case: no reason why it should be (TODO: open arrow issue)

AttributeError: 'pyarrow.lib.ChunkedArray' object has no attribute 'offset'

# Fixed code
def ensure_not_chunked(arrow_array):
    if isinstance(arrow_array, pa.ChunkedArray):
        if len(arrow_array.chunks) == 1:
            return arrow_array.chunks[0]
        table = pa.Table.from_arrays([arrow_array], ["single"])
        table_concat = table.combine_chunks()
        column = table_concat.columns[0]
        assert column.num_chunks == 1
        arrow_array = column.chunk(0)
    return arrow_array