Validated by test runs on GPU/Kubernetes, LGTM
@@ -6,16 +6,34 @@ converter_method_dict = {
'ecfp': concat_mols,
'nfp': concat_mols,
'nfp_gwm': concat_mols,
+ 'nfp_wle': concat_mols,
+ 'nfp_cwle': concat_mols,
+ 'nfp_gwle': concat_mols,
'ggnn': concat_mols,
'ggnn_gwm': concat_mols,
+ 'ggnn_wle': concat_mols,
+ 'ggnn_cwle': concat_mols,
+ 'ggnn_gwle': concat_mols,
'gin': concat_mols,
'gin_gwm': concat_mols,
+ 'gin_wle': concat_mols,
+ 'gin_cwle': concat_mols,
+ 'gin_gwle': concat_mols,
'schnet': concat_mols,
'weavenet': concat_mols,
'relgcn': concat_mols,
+ 'relgcn_wle': concat_mols,
+ 'relgcn_cwle': concat_mols,
+ 'relgcn_gwle': concat_mols,
'rsgcn': concat_mols,
'rsgcn_gwm': concat_mols,
+ 'rsgcn_wle': concat_mols,
+ 'rsgcn_cwle': concat_mols,
+ 'rsgcn_gwle': concat_mols,
'relgat': concat_mols,
+ 'relgat_wle': concat_mols,
+ 'relgat_cwle': concat_mols,
+ 'relgat_gwle': concat_mols,
'gnnfilm': concat_mols,
'megnet': megnet_converter,
'cgcnn': cgcnn_converter
