metnet 2 import issue

[tarunluthra]

I am facing issues after installing Metnet2, I am running on python 3.9.2 in a conda environment.

Bug Report

Describe the Bug

Attempting to import MetNet and MetNet2 from the metnet package results in a ModuleNotFoundError. Specifically, the error indicates that the module metnet.layers cannot be found:

ModuleNotFoundError: No module named 'metnet.layers'; 'metnet' is not a package

This error occurs when attempting to run the following code:

from metnet import MetNet, MetNet2

# Load pre-trained models
model = MetNet().from_pretrained("openclimatefix/metnet")
model = MetNet2().from_pretrained("openclimatefix/metnet-2")

To Reproduce

Steps to reproduce the behavior:

1. Ensure that the metnet repository is cloned and installed.
2. Open a Python environment within the project directory.
3. Run the import statements and model loading code:

   from metnet import MetNet, MetNet2
   model = MetNet().from_pretrained("openclimatefix/metnet")
   model = MetNet2().from_pretrained("openclimatefix/metnet-2")

4. Observe the ModuleNotFoundError.

Expected Behavior

The expected behavior is that the MetNet and MetNet2 classes are successfully imported, and the from_pretrained method loads pre-trained models without errors.

Additional Context

• Error Message:

  ModuleNotFoundError: No module named 'metnet.layers'; 'metnet' is not a package

can anyone help me with this?