Closed genec1 closed 4 years ago
Hi,
In our original code, molecule.y is always a list, so that one can always call molecule.y[0].
This if statement that you show here differentiates between multi-task (where molecule.y[0] is np.ndarray) and single-task setting (where molecule.y[0] is a scalar).
I've modified the data_loader code and am running into a crash in
featurization.data_utils.mol_collate_func
here:I didn't debug the original code to see why it doesn't crash here, but with my code
molecule.y
is a scalar, somolecule.y[0]
doesn't work. It seems like it should be: