From 196936df5cb638a98f3344df4360730208966cec Mon Sep 17 00:00:00 2001 From: "UnravelSports [JB]" Date: Mon, 22 Jul 2024 13:53:07 +0200 Subject: [PATCH] forgot --- unravel/utils/objects/custom_spektral_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unravel/utils/objects/custom_spektral_dataset.py b/unravel/utils/objects/custom_spektral_dataset.py index 6a50099..dad8292 100644 --- a/unravel/utils/objects/custom_spektral_dataset.py +++ b/unravel/utils/objects/custom_spektral_dataset.py @@ -55,7 +55,7 @@ def __convert(self, data) -> List[Graph]: return [g.to_spektral_graph() for g in self.data] elif isinstance(data[0], dict): return [ - SpektralGraph(x=g["x"], a=g["a"], e=g["e"], y=g["y"], id=g["id"]) + Graph(x=g["x"], a=g["a"], e=g["e"], y=g["y"], id=g["id"]) for g in data ] else: