mlflow.genai

mlflow.genai.create_dataset(uc_table_name: str, experiment_id: Optional[Union[str, list[str]]] = None) → EvaluationDataset[source]

Create a dataset with the given name and associate it with the given experiment.

Parameters

uc_table_name – The UC table name of the dataset.
experiment_id – The ID of the experiment to associate the dataset with. If not provided, the current experiment is inferred from the environment.

Returns

The created dataset.

Return type

EvaluationDataset

mlflow.genai.delete_dataset(uc_table_name: str) → None[source]

Delete the dataset with the given name.

mlflow.genai.get_dataset(uc_table_name: str) → EvaluationDataset[source]

Get the dataset with the given name.