From 40c08935e0933c235d8c008ec7bd79b14018304e Mon Sep 17 00:00:00 2001
From: Markus Semmler <markus.semmler@unternehmertum.de>
Date: Fri, 22 Dec 2023 19:21:52 +0100
Subject: [PATCH] Fix error in log.py

---
 src/re_classwise_shapley/log.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/src/re_classwise_shapley/log.py b/src/re_classwise_shapley/log.py
index cb8762be..2529fe6c 100644
--- a/src/re_classwise_shapley/log.py
+++ b/src/re_classwise_shapley/log.py
@@ -48,17 +48,15 @@ def log_datasets(datasets: pd.DataFrame):
         for dataset_type in ["val_set", "test_set"]:
             dataset = row[dataset_type]
             dataset_name = row["dataset_name"]
-            repetition_id = str(row["repetition_id"])
             mlflow.log_input(
                 mlflow.data.pandas_dataset.from_pandas(
                     dataset_to_dataframe(dataset),
                     targets=dataset.target_names[0],
-                    name=f"{dataset_name}_{repetition_id}_{dataset_type}",
+                    name=f"{dataset_name}_{dataset_type}",
                 ),
                 tags={
                     "set": dataset_type,
                     "dataset": dataset_name,
-                    "repetition": repetition_id,
                 },
             )