gnn fix

aditya0by0 · aditya0by0 · commit 46a1ed6eb2f9 · 2026-02-03T11:11:45.000+01:00
diff --git a/chebifier/prediction_models/gnn_predictor.py b/chebifier/prediction_models/gnn_predictor.py
@@ -10,5 +10,5 @@ def __init__(
     ):
         super().__init__(model_name, ckpt_path, **kwargs)
         print(
-            f"Initialised GNN model {self.model_name} (device: {self._predictor.device})"
+            f"Initialised GNN model {self.model_name} (device: {self.predictor.device})"
         )

Original file line number	Diff line number	Diff line change
`@@ -10,5 +10,5 @@ def __init__(`
`10`	`10`	`):`
`11`	`11`	`super().__init__(model_name, ckpt_path, **kwargs)`
`12`	`12`	`print(`
`13`		`- f"Initialised GNN model {self.model_name} (device: {self._predictor.device})"`
	`13`	`+ f"Initialised GNN model {self.model_name} (device: {self.predictor.device})"`
`14`	`14`	`)`