We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 79fc500 commit d2fe1d2Copy full SHA for d2fe1d2
chebai_graph/preprocessing/reader/static_gni.py
@@ -40,9 +40,12 @@ def name(self) -> str:
40
41
def _read_data(self, raw_data):
42
data: GeomData = super()._read_data(raw_data)
43
+ if data is None:
44
+ return None
45
+
46
random_x = torch.empty(data.x.shape[0], self.num_node_properties)
47
random_edge_attr = torch.empty(
- data.edge_index.shape[1], self.num_bond_properties
48
+ data.edge_attr.shape[0], self.num_bond_properties
49
)
50
random_molecule_properties = torch.empty(1, self.num_molecule_properties)
51
0 commit comments