Fixed missing logging function export (#700)

* Added export of `initialize_mlflow` * Updated Changelog * Ran pre-commit
NVIDIA · Oct 28, 2024 · 29774fe · 29774fe
1 parent 89fae50
commit 29774fe
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -16,4 +16,5 @@
 
 from .console import PythonLogger, RankZeroLoggingWrapper
 from .launch import LaunchLogger
+from .mlflow import initialize_mlflow
 from .wandb import initialize_wandb