fix gat import error

Author: aditya0by0

chebifier/model_registry.py

@@ -15,7 +15,6 @@
     ChemlogOrganoXCompoundPredictor,
     ChemlogXMolecularEntityPredictor,
 )
-from chebifier.prediction_models.gnn_predictor import GATPredictor
 
 ENSEMBLES = {
     "mv": BaseEnsemble,
@@ -27,7 +26,7 @@
 MODEL_TYPES = {
     "electra": ElectraPredictor,
     "resgated": GNNPredictor,
-    "gat": GATPredictor,
+    "gat": GNNPredictor,
     "chemlog": ChemlogAllPredictor,
     "chemlog_peptides": ChemlogPeptidesPredictor,
     "chebi_lookup": ChEBILookupPredictor,