Registry for trainer_name_mapping is empty

Hi,
I am trying to optimize a surface using equiformer_v2 as follows;
config_yml_path = r’equiformer_v2_N@20_L@6_M@3_153M.yml’
checkpoint_path = r’eq2_153M_ec4_allmd.pt’
calc = OCPCalculator(config_yml= config_yml_path ,checkpoint_path=checkpoint_path)
atom.calc = calc
However, I keep getting this error “RuntimeError: Registry for trainer_name_mapping is empty. You may have forgot to load a module.” → traced back to the registry.py
Could you please help me with this issue?