add gat model type

Author: sfluegel05

chebifier/model_registry.py

@@ -14,6 +14,7 @@
     ChemlogOrganoXCompoundPredictor,
     ChemlogXMolecularEntityPredictor,
 )
+from chebifier.prediction_models.gnn_predictor import GATPredictor
 
 ENSEMBLES = {
     "mv": BaseEnsemble,
@@ -25,6 +26,7 @@
 MODEL_TYPES = {
     "electra": ElectraPredictor,
     "resgated": ResGatedPredictor,
+    "gat": GATPredictor,
     "chemlog_peptides": ChemlogPeptidesPredictor,
     "chebi_lookup": ChEBILookupPredictor,
     "chemlog_element": ChemlogXMolecularEntityPredictor,