search

NVIDIA / modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods
https://developer.nvidia.com/modulus
Apache License 2.0
795 stars 172 forks source link

🐛[BUG]: need to install termcolor and mlflow to use ERA5DataPipe #489

Open nbren12 opened 1 month ago

nbren12 commented 1 month ago

Version

0.6.0

On which installation method(s) does this occur?

Pip

Describe the issue

----> 1 from  modulus.datapipes.climate.era5_hdf5 import ERA5HDF5Datapipe

File /usr/local/lib/python3.10/dist-packages/modulus/datapipes/climate/__init__.py:17
      1 # SPDX-FileCopyrightText: Copyright (c) 2023 - 2024 NVIDIA CORPORATION & AFFILIATES.
      2 # SPDX-FileCopyrightText: All rights reserved.
      3 # SPDX-License-Identifier: Apache-2.0
   (...)
     14 # See the License for the specific language governing permissions and
     15 # limitations under the License.
---> 17 from .climate import ClimateDatapipe, ClimateDataSourceSpec
     18 from .era5_hdf5 import ERA5HDF5Datapipe

File /usr/local/lib/python3.10/dist-packages/modulus/datapipes/climate/climate.py:47
     45 from modulus.datapipes.datapipe import Datapipe
     46 from modulus.datapipes.meta import DatapipeMetaData
---> 47 from modulus.launch.logging import PythonLogger
     49 Tensor = torch.Tensor
     52 @dataclass
     53 class MetaData(DatapipeMetaData):

File /usr/local/lib/python3.10/dist-packages/modulus/launch/logging/__init__.py:19
     17 from .console import PythonLogger, RankZeroLoggingWrapper
     18 from .launch import LaunchLogger
---> 19 from .mlflow import initialize_mlflow
     20 from .wandb import initialize_wandb

File /usr/local/lib/python3.10/dist-packages/modulus/launch/logging/mlflow.py:22
     19 from pathlib import Path
     20 from typing import Literal, Tuple
---> 22 import mlflow
     23 import torch
     24 from mlflow.entities.run import Run

Minimum reproducible example

No response

Relevant log output

No response

Environment details

No response