def _make_columns(
specs: TreeDict[tf.TypeSpec],
ds_info: Optional[dataset_info.DatasetInfo],
) -> List[ColumnInfo]:
"""Extract the columns info of the `panda.DataFrame`."""
return [
ColumnInfo.from_spec(path, ds_info)
for path, _ in py_utils.flatten_with_path(specs)
]
def _flatten_with_path(v):
return list(py_utils.flatten_with_path(v))
