Optimizer: Key error in column pruning

[ericpai]

Source code

df = md.DataFrame({"A": [1, 2, 3], "B": [1, 2, 3], "D": [1,2,3]})
df_category = df["A"]
df["C"] = df_category.case_when([(df_category == 1, 10)])
df.execute().fetch()

Error report:

    *[self._context[successor][data] for successor in successors]
      ~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^
KeyError: Series <op=DataFrameIndex, key=c9db36f5dbdc13c3de25f6dabd4fb775_0>

[ericpai]

Fixed in v1.0.0rc2