From c891cc679d5e3ab75e1cf7726f81a2dbfcc47a48 Mon Sep 17 00:00:00 2001 From: Jan Meyer <20856376+jan-meyer-1986@users.noreply.github.com> Date: Mon, 25 Nov 2024 13:08:58 +0100 Subject: [PATCH] return correct population table name --- mlflow/getml/autologging.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlflow/getml/autologging.py b/mlflow/getml/autologging.py index e6531d68563ca..1216b7aaa93f0 100644 --- a/mlflow/getml/autologging.py +++ b/mlflow/getml/autologging.py @@ -223,7 +223,7 @@ def _log_pretraining_metadata(autologging_client: MlflowAutologgingQueueingClien if log_datasets: try: datasets = [] - population_dataset: PandasDataset = mlflow.data.from_pandas(args[0].population.to_pandas(), name = args[0].population.name) #args[0].population.name returns the wrong name + population_dataset: PandasDataset = mlflow.data.from_pandas(args[0].population.to_pandas(), name = args[0].population.base.name) tags = [InputTag(key=MLFLOW_DATASET_CONTEXT, value='Population')] datasets.append(DatasetInput(dataset=population_dataset._to_mlflow_entity(), tags=tags))