TorchDrug can't use Lr_Scheduler

[mrzzmrzz]

Hey, I found a bug that when I load the related TorchDrug modules, I can't use the torch.optim.lr_scheduler.
Look at this picture which comes from the TorchDrug Colab files(Property Prediction). I add one lr_schduler for the optimizer. and it throws an error.

However, When I don't load any TorchDrug modules, I can use the optimizer normally.

[mrzzmrzz]

OK, Seems like that the bug is also mentioned in #148. I also tried the MultiStepLr

Also Throw Error.

[mrzzmrzz]

CosineAnnealingLR

[Oxer11]

Hi! You're right. This issue has been reported in #148 and fixed in 4ea99a0. You can clone the latest repo for usage.

Sorry for the inconvenience. We'll update the conda and pip versions later.

[mrzzmrzz]

OK, Thank you For your timely repair work!