def _as_df(ds_name: str, **kwargs) -> pandas.DataFrame:
"""Loads the dataset as `pandas.DataFrame`."""
with testing.mock_data(num_examples=3):
ds, ds_info = load.load(ds_name,
split='train',
with_info=True,
**kwargs)
df = as_dataframe.as_dataframe(ds, ds_info)
return df
def test_as_dataframe():
"""Tests that as_dataframe works without the `tfds.core.DatasetInfo`."""
ds = tf.data.Dataset.from_tensor_slices({
'some_key': [1, 2, 3],
'nested': {
'sub1': [1.0, 2.0, 3.0],
},
})
df = as_dataframe.as_dataframe(ds)
assert isinstance(df, pandas.DataFrame)
assert df._repr_html_().startswith('<style')