From 29774fe08a1d3a324d21852452826f5ccdbf1e32 Mon Sep 17 00:00:00 2001
From: Alexander Kiefer <56556451+alexk101@users.noreply.github.com>
Date: Mon, 28 Oct 2024 15:41:04 -0700
Subject: [PATCH] Fixed missing logging function export (#700)

* Added export of `initialize_mlflow`

* Updated Changelog

* Ran pre-commit
---
 CHANGELOG.md                       | 1 +
 modulus/launch/logging/__init__.py | 1 +
 2 files changed, 2 insertions(+)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 9bddb143c6..a6ebcfffd0 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -25,6 +25,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Fixed timezone calculation in datapipe cosine zenith utility.
 - Refactored EDMPrecondSRV2 preconditioner and fixed the bug related to the metadata
 - Extended the checkpointing utility to store metadata.
+- Corrected missing export of loggin function used by transolver model
 
 ### Deprecated
 
diff --git a/modulus/launch/logging/__init__.py b/modulus/launch/logging/__init__.py
index fe4fa83db7..50466a3678 100644
--- a/modulus/launch/logging/__init__.py
+++ b/modulus/launch/logging/__init__.py
@@ -16,4 +16,5 @@
 
 from .console import PythonLogger, RankZeroLoggingWrapper
 from .launch import LaunchLogger
+from .mlflow import initialize_mlflow
 from .wandb import initialize_wandb