# Buggy code
def __call__(self, df):
    if self.col_names is not None:
        # if col_names is a list, return a DataFrame, else return a Series
        if isinstance(self._col_names, list):
            dtypes = df.dtypes[self._col_names]
            columns = parse_index(pd.Index(self._col_names), store_data=True)
            return self.new_dataframe(
                [df],
                shape=(df.shape[0], len(self._col_names)),
                dtypes=dtypes,
                index_value=df.index_value,
                columns_value=columns,
            )
        else:
            dtype = df.dtypes[self._col_names]
            return self.new_series(
                [df],
                shape=(df.shape[0],),
                dtype=dtype,
                index_value=df.index_value,
                name=self._col_names,
            )
    else:
        if isinstance(self.mask, (SERIES_TYPE, DATAFRAME_TYPE)):
            index_value = parse_index(
                pd.Index(
                    [], dtype=df.index_value.to_pandas().dtype, name=df.index_value.name
                ),
                df,
                self._mask,
            )
            return self.new_dataframe(
                [df, self._mask],
                shape=(np.nan, df.shape[1]),
                dtypes=df.dtypes,
                index_value=index_value,
                columns_value=df.columns_value,
            )
        else:
            index_value = parse_index(
                pd.Index(
                    [], dtype=df.index_value.to_pandas().dtype, name=df.index_value.name
                ),
                df,
                self._mask,
            )
            return self.new_dataframe(
                [df],
                shape=(np.nan, df.shape[1]),
                dtypes=df.dtypes,
                index_value=index_value,
                columns_value=df.columns_value,
            )


# Traceback (TypeError)
Traceback (most recent call last):
File "/Users/wenjun.swj/miniconda3/lib/python3.8/site-packages/IPython/core/interactiveshell.py", line 3417, in run_code
exec(code_obj, self.user_global_ns, self.user_ns)
File "<ipython-input-9-60fd09690beb>", line 1, in <module>
df[df[0] > 0.5].execute()
File "/Users/wenjun.swj/Code/mars/mars/core.py", line 643, in execute
return run()
File "/Users/wenjun.swj/Code/mars/mars/core.py", line 639, in run
self.data.execute(session, **kw)
File "/Users/wenjun.swj/Code/mars/mars/core.py", line 379, in execute
return run()
File "/Users/wenjun.swj/Code/mars/mars/core.py", line 374, in run
session.run(self, **kw)
File "/Users/wenjun.swj/Code/mars/mars/session.py", line 505, in run
result = self._sess.run(*tileables, **kw)
File "/Users/wenjun.swj/Code/mars/mars/session.py", line 111, in run
res = self._executor.execute_tileables(tileables, **kw)
File "/Users/wenjun.swj/Code/mars/mars/utils.py", line 451, in _inner
return func(*args, **kwargs)
File "/Users/wenjun.swj/Code/mars/mars/executor.py", line 860, in execute_tileables
chunk_graph = chunk_graph_builder.build(
File "/Users/wenjun.swj/Code/mars/mars/utils.py", line 451, in _inner
return func(*args, **kwargs)
File "/Users/wenjun.swj/Code/mars/mars/tiles.py", line 347, in build
chunk_graph = super().build(
File "/Users/wenjun.swj/Code/mars/mars/utils.py", line 451, in _inner
return func(*args, **kwargs)
File "/Users/wenjun.swj/Code/mars/mars/tiles.py", line 262, in build
self._on_tile_failure(tileable_data.op, exc_info)
File "/Users/wenjun.swj/Code/mars/mars/tiles.py", line 301, in inner
raise exc_info[1].with_traceback(exc_info[2]) from None
File "/Users/wenjun.swj/Code/mars/mars/tiles.py", line 242, in build
tiled = self._tile(tileable_data, tileable_graph)
File "/Users/wenjun.swj/Code/mars/mars/tiles.py", line 337, in _tile
return super()._tile(tileable_data, tileable_graph)
File "/Users/wenjun.swj/Code/mars/mars/tiles.py", line 201, in _tile
tds[0]._inplace_tile()
File "/Users/wenjun.swj/Code/mars/mars/core.py", line 168, in _inplace_tile
return handler.inplace_tile(self)
File "/Users/wenjun.swj/Code/mars/mars/tiles.py", line 136, in inplace_tile
dispatched = self.dispatch(to_tile.op)
File "/Users/wenjun.swj/Code/mars/mars/utils.py", line 451, in _inner
return func(*args, **kwargs)
File "/Users/wenjun.swj/Code/mars/mars/tiles.py", line 119, in dispatch
tiled = op_cls.tile(op)
File "/Users/wenjun.swj/Code/mars/mars/dataframe/indexing/getitem.py", line 273, in tile
return cls.tile_with_mask(op)
File "/Users/wenjun.swj/Code/mars/mars/dataframe/indexing/getitem.py", line 287, in tile_with_mask
align_dataframe_series(in_df, mask, axis='index')
File "/Users/wenjun.swj/Code/mars/mars/dataframe/align.py", line 713, in align_dataframe_series
index_splits, index_nsplits = _calc_axis_splits(left.index_value, right.index_value,
File "/Users/wenjun.swj/Code/mars/mars/dataframe/align.py", line 490, in _calc_axis_splits
right_splits = left_splits = [[c] for c in left_chunk_index_min_max]
TypeError: 'NoneType' object is not iterable

# Fixed code
def __call__(self, df):
    if self.col_names is not None:
        # if col_names is a list, return a DataFrame, else return a Series
        if isinstance(self._col_names, list):
            dtypes = df.dtypes[self._col_names]
            columns = parse_index(pd.Index(self._col_names), store_data=True)
            return self.new_dataframe(
                [df],
                shape=(df.shape[0], len(self._col_names)),
                dtypes=dtypes,
                index_value=df.index_value,
                columns_value=columns,
            )
        else:
            dtype = df.dtypes[self._col_names]
            return self.new_series(
                [df],
                shape=(df.shape[0],),
                dtype=dtype,
                index_value=df.index_value,
                name=self._col_names,
            )
    else:
        if isinstance(self.mask, (SERIES_TYPE, DATAFRAME_TYPE, TENSOR_TYPE)):
            index_value = parse_index(
                pd.Index(
                    [], dtype=df.index_value.to_pandas().dtype, name=df.index_value.name
                ),
                df,
                self._mask,
            )
            return self.new_dataframe(
                [df, self._mask],
                shape=(np.nan, df.shape[1]),
                dtypes=df.dtypes,
                index_value=index_value,
                columns_value=df.columns_value,
            )
        else:
            index_value = parse_index(
                pd.Index(
                    [], dtype=df.index_value.to_pandas().dtype, name=df.index_value.name
                ),
                df,
                self._mask,
            )
            return self.new_dataframe(
                [df],
                shape=(np.nan, df.shape[1]),
                dtypes=df.dtypes,
                index_value=index_value,
                columns_value=df.columns_value,
            )