Waterbirds give 0 worst-group accuracy #148

Training waterbirds dataset out-of-the-box gives 0 worst group accuracy. Digging deeper, I noticed that all the predictions immediately become the 0 label. Any advice would be helpful. Thanks in advance.