From 26fd4d3c23856b55d8005b78217f1ac7ea299228 Mon Sep 17 00:00:00 2001
From: Jesse Chan <jesse.chan@databricks.com>
Date: Fri, 30 Aug 2024 09:09:21 +0900
Subject: [PATCH] Remove step from log image key

Signed-off-by: Jesse Chan <jesse.chan@databricks.com>
---
 composer/loggers/mlflow_logger.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/composer/loggers/mlflow_logger.py b/composer/loggers/mlflow_logger.py
index 7a8acdd511..eb2f917a92 100644
--- a/composer/loggers/mlflow_logger.py
+++ b/composer/loggers/mlflow_logger.py
@@ -601,7 +601,7 @@ def log_images(
                 assert isinstance(self._run_id, str)
                 self._mlflow_client.log_image(
                     image=image,
-                    key=f'{name}_{step}_{im_ind}',
+                    key=f'{name}_{im_ind}',
                     run_id=self._run_id,
                     step=step,
                 )