🐛[BUG]: Cannot import modulus.launch.logging

[simonbyrne]

Version

0.9.0

On which installation method(s) does this occur?

Docker

Describe the issue

If I attempt to import modulus.launch.logging, I get an error that MLFlow is not installed (note, I am not attempting to load modulus.launch.logging.mlflow)

The problem appears to be that the __init__.py loads mlflow as a submodule:

modulus/modulus/launch/logging/__init__.py

Line 19 in f607d8f

from .mlflow import initialize_mlflow

Minimum reproducible example

Using the current NGC container (24.12)

import modulus.launch.logging

Relevant log output

Traceback (most recent call last):
  File "/usr/local/lib/python3.10/dist-packages/modulus/launch/logging/mlflow.py", line 25, in <module>
    import mlflow  # noqa: F401 for docs
ModuleNotFoundError: No module named 'mlflow'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/local/lib/python3.10/dist-packages/modulus/launch/logging/__init__.py", line 19, in <module>
    from .mlflow import initialize_mlflow
  File "/usr/local/lib/python3.10/dist-packages/modulus/launch/logging/mlflow.py", line 29, in <module>
    raise ImportError(
ImportError: These utilities require the MLFlow library. Install MLFlow using `pip install mlflow`. For more info, refer: https://www.mlflow.org/docs/2.5.0/quickstart.html#install-mlflow

Environment details