From 7d34d66e261e7f2c16686e2e03bdcb4643b6f4ec Mon Sep 17 00:00:00 2001
From: Mihir Patel <mihir.v.patel7@gmail.com>
Date: Wed, 4 Sep 2024 14:27:28 -0700
Subject: [PATCH] Remove step in log_image for MLFlow (#3601)

---
 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,
                 )