aswin-raghavan commited on
Commit
fc18f1c
·
1 Parent(s): e23ed18

bugfix unique

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -91,7 +91,7 @@ def quantize_embeds(embeds):
91
  def update_exemplars(df, rng, exemplars, lut):
92
  embeds = df[['image_embed']].to_numpy()
93
  labels = df[['label']].to_numpy()
94
- assert np.all(labels.unique() == [0, 1])
95
  labels_zero_idx = (labels == 0).nonzero()
96
  labels_one_idx = (labels == 1).nonzero()
97
  print(labels_zero_idx.shape, " zeros and ", labels_one_idx.shape, " ones")
 
91
  def update_exemplars(df, rng, exemplars, lut):
92
  embeds = df[['image_embed']].to_numpy()
93
  labels = df[['label']].to_numpy()
94
+ assert np.all(np.unique(labels) == [0, 1])
95
  labels_zero_idx = (labels == 0).nonzero()
96
  labels_one_idx = (labels == 1).nonzero()
97
  print(labels_zero_idx.shape, " zeros and ", labels_one_idx.shape, " ones")