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))