Using data getter on empty DataGrid produces key error

[ollyhensby]

Describe the bug
When there's no data in the grid, the data getter produces an error.

To Reproduce
Steps to reproduce the behavior:

1. Import DataGrid and pandas

from ipydatagrid import DataGrid
import pandas as pd

2. Create simple dataframe and input into DataGrid

df = pd.DataFrame([1], columns=["a"])
grid = DataGrid(df)
grid

We get this output:

We can also call the data getter:

grid.data

3. Now drop all the rows in the dataframe and set using the data setter.

grid.data = grid.data.drop(0)

We see that the DataGrid UI object reflects those changes (we only have the columns):

4. Now call the data getter:

grid.data

5. See error

Expected behavior
I expect an empty dataframe to be returned like this when calling the data getter:

Environment (please complete the following information):

• Operating System and Version: Ubuntu 20.04 WSL
• Browser [e.g. Firefox, Safari] (if applicable): Chrome (Using Jupyter Lab)

Any help with this is much appreciated. Cheers!

[ibdafna]

Hello 👋 @ollyhensby and welcome to the ipydatagrid community! Many thanks for reporting - I'll get this fixed!

[ibdafna]

Closed via #358 Awesome work @ollyhensby 🙏 Thank you!