[molflux.features.plugins.physicsml]
allegro = physicsml.models.allegro.allegro_prism:Allegro
ani = physicsml.models.ani.ani_prism:ANI
egnn = physicsml.models.egnn.egnn_prism:EGNN
mace = physicsml.models.mace.mace_prism:MACE
nequip = physicsml.models.nequip.nequip_prism:Nequip
physicsml_features = physicsml.featurisation.physicsml_features.physicsml_features:PhysicsMLFeatures

[molflux.modelzoo.plugins.physicsml]
adapter_egnn_model = physicsml.models.egnn.adapter.adapter_egnn_model:AdapterEGNNModel
adapter_mace_model = physicsml.models.mace.adapter.adapter_mace_model:AdapterMACEModel
adapter_nequip_model = physicsml.models.nequip.adapter.adapter_nequip_model:AdapterNequipModel
allegro_model = physicsml.models.allegro.supervised.allegro_model:AllegroModel
ani_model = physicsml.models.ani.supervised.ani_model:ANIModel
egnn_model = physicsml.models.egnn.supervised.egnn_model:EGNNModel
ensemble_ani_model = physicsml.models.ani.ensemble.ensemble_ani_model:EnsembleANIModel
mace_model = physicsml.models.mace.supervised.mace_model:MACEModel
mean_var_allegro_model = physicsml.models.allegro.mean_var.mean_var_allegro_model:MeanVarAllegroModel
mean_var_ani_model = physicsml.models.ani.mean_var.mean_var_ani_model:MeanVarANIModel
mean_var_egnn_model = physicsml.models.egnn.mean_var.mean_var_egnn_model:MeanVarEGNNModel
mean_var_mace_model = physicsml.models.mace.mean_var.mean_var_mace_model:MeanVarMACEModel
mean_var_nequip_model = physicsml.models.nequip.mean_var.mean_var_nequip_model:MeanVarNequipModel
nequip_model = physicsml.models.nequip.supervised.nequip_model:NequipModel
ssf_egnn_model = physicsml.models.egnn.ssf.ssf_egnn_model:SsfEGNNModel
ssf_mace_model = physicsml.models.mace.ssf.ssf_mace_model:SsfMACEModel
ssf_nequip_model = physicsml.models.nequip.ssf.ssf_nequip_model:SsfNequipModel
tensor_net_model = physicsml.models.tensor_net.supervised.tensor_net_model:TensorNetModel
