fix model instantiation

Author: sfluegel05

chebifier/prediction_models/electra_predictor.py

@@ -53,7 +53,6 @@ def init_model(self, ckpt_path: str, **kwargs) -> "Electra":
             map_location=self.device,
             criterion=None,
             strict=False,
-            metrics=dict(train=dict(), test=dict(), validation=dict()),
             pretrained_checkpoint=None,
         )
         model.eval()

chebifier/prediction_models/gnn_predictor.py

@@ -68,8 +68,6 @@ def init_model(self, ckpt_path: str, **kwargs) -> "ResGatedGraphPred":
             map_location=torch.device(self.device),
             criterion=None,
             strict=False,
-            metrics=dict(train=dict(), test=dict(), validation=dict()),
-            pretrained_checkpoint=None,
         )
         model.eval()
         return model
@@ -113,8 +111,6 @@ def init_model(self, ckpt_path: str, **kwargs) -> "GATGraphPred":
             map_location=torch.device(self.device),
             criterion=None,
             strict=False,
-            metrics=dict(train=dict(), test=dict(), validation=dict()),
-            pretrained_checkpoint=None,
         )
         model.eval()
         return model