PyTorch Libary Import Error (related to new Pytorch release?)

[branlsnyder]

Hi acids,

Thanks for setting up this open repo. Excited to dive in deeper with all of this.

I'm following cli_helper.py, and when running train_rave.py, I receive the message:

'''
ImportError: cannot import name 'get_num_classes' from 'torchmetrics.utilities.data' (/opt/anaconda3/lib/python3.9/site-packages/torchmetrics/utilities/data.py)
''''

Is this related to a recent release of PyTorchLightning, where they removed 'get_num_classes'? (commit here)

What would you suggest is the best steps for me to take? Thanks.
EDIT: I installed an older version of torchmetrics, and this has solved my issue for now.

Here is the full text of my terminal:
'''
(base) Brandon-MacBook-Pro:RAVE BrandonSnyder$ python train_rave.py --name second_training_test_01 --wav /Users/BrandonSnyder/Dropbox/Documents (Dbox)/GitHub/RAVE/first_training_01-outfiles --preprocessed /Users/BrandonSnyder/Dropbox/Documents (Dbox)/GitHub/RAVE/first_training_01-tmp/second_training_test_01/rave
Traceback (most recent call last):
File "/Users/BrandonSnyder/Dropbox/Documents (Dbox)/GitHub/RAVE/train_rave.py", line 4, in
from rave.model import RAVE
File "/Users/BrandonSnyder/Dropbox/Documents (Dbox)/GitHub/RAVE/rave/init.py", line 1, in
from .model import RAVE
File "/Users/BrandonSnyder/Dropbox/Documents (Dbox)/GitHub/RAVE/rave/model.py", line 5, in
import pytorch_lightning as pl
File "/opt/anaconda3/lib/python3.9/site-packages/pytorch_lightning/init.py", line 20, in
from pytorch_lightning import metrics # noqa: E402
File "/opt/anaconda3/lib/python3.9/site-packages/pytorch_lightning/metrics/init.py", line 15, in
from pytorch_lightning.metrics.classification import ( # noqa: F401
File "/opt/anaconda3/lib/python3.9/site-packages/pytorch_lightning/metrics/classification/init.py", line 14, in
from pytorch_lightning.metrics.classification.accuracy import Accuracy # noqa: F401
File "/opt/anaconda3/lib/python3.9/site-packages/pytorch_lightning/metrics/classification/accuracy.py", line 18, in
from pytorch_lightning.metrics.utils import deprecated_metrics, void
File "/opt/anaconda3/lib/python3.9/site-packages/pytorch_lightning/metrics/utils.py", line 22, in
from torchmetrics.utilities.data import get_num_classes as _get_num_classes
ImportError: cannot import name 'get_num_classes' from 'torchmetrics.utilities.data' (/opt/anaconda3/lib/python3.9/site-packages/torchmetrics/utilities/data.py)
(base) Brandon-MacBook-Pro:RAVE BrandonSnyder$
'''

[caillonantoine]

This is fixed in the latest version of the repo :)

[branlsnyder]

super! thanks Antoine