Bug in main.py for OptimConfig in new versions of PyG

[GARV-k]

my package details are : PyG-2.5.2 torch - 2.1.0 CUDA Version: 12.2 Python : 3.9.19

In the main.py file, I just imported the OptimConfig and SchedulerConfig from torch_geometric.graphgym.optim. it gives a error : Traceback (most recent call last): File "/home/iplab/garv_iit-bhu/GraphTransformer/lrgb/main.py", line 178, in optimizer = create_optimizer(model.parameters(), File "/home/iplab/anaconda3/envs/lrgb/lib/python3.9/site-packages/torch_geometric/graphgym/optim.py", line 38, in create_optimizer return from_config(func)(params, cfg=cfg) File "/home/iplab/anaconda3/envs/lrgb/lib/python3.9/site-packages/torch_geometric/graphgym/config.py", line 578, in wrapper raise ValueError(f"'cfg.{arg_name}' undefined") ValueError: 'cfg.optimizer_config' undefined

Please help me out. Also let me know if u need any more details.