Fixed edge type bug when loss function is called in hamming_loss_augmentented_inference