From 856eb9f22926df4a0e07c241800dd5bbf1ffc6cb Mon Sep 17 00:00:00 2001 From: Shivam Shukla Date: Tue, 7 Nov 2023 03:45:19 +0530 Subject: [PATCH] Fixed the wrong argument name in 'load_molecules' function. --- training_data.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/training_data.py b/training_data.py index 3e70973..da531c7 100644 --- a/training_data.py +++ b/training_data.py @@ -75,7 +75,7 @@ def generate_z_values(batch_size=32, z_dim=32, vertexes=32, b_dim=32, m_dim=32, # return drug_graphs, real_graphs, a_tensor, x_tensor, drugs_a_tensor, drugs_x_tensor -def load_molecules(batch=None, b_dim=32, m_dim=32, device=None, batch_size=32): +def load_molecules(data=None, b_dim=32, m_dim=32, device=None, batch_size=32): data = data.to(device) a = geoutils.to_dense_adj( edge_index = data.edge_index, @@ -90,4 +90,4 @@ def load_molecules(batch=None, b_dim=32, m_dim=32, device=None, batch_size=32): x_tensor_vec = x_tensor.reshape(batch_size,-1) real_graphs = torch.concat((x_tensor_vec,a_tensor_vec),dim=-1) - return real_graphs, a_tensor, x_tensor \ No newline at end of file + return real_graphs, a_tensor, x_tensor