v3.0.2

• fix wandb logging bug where if mask is all NaNs, the returned value was not a tuple but a float (i.e. NaN)
• ensure committee functionality is robust towards datasets with varying number of atoms