return correct population table name

getml · Nov 25, 2024 · c891cc6 · c891cc6
1 parent 4797edc
commit c891cc6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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))