Made mnist_label_source generic.