nemo_curator.utils.column_utils

View as Markdown

Module Contents

Functions

NameDescription
resolve_filename_columnResolve the filename column name based on the input parameter.

API

nemo_curator.utils.column_utils.resolve_filename_column(
add_filename_column: bool | str
) -> str | None

Resolve the filename column name based on the input parameter.

Parameters:

add_filename_column
bool | str

Can be:

  • True: Use default filename column name
  • False: No filename column
  • str: Use the provided string as column name

Returns: str | None

str | None: The filename column name or None if not needed