message: values returns a dict so we need to change it into a list
diff: @@ -57,9 +57,11 @@ def assert_dataset_origin(keras_dataset, federated_dataset):
  for (x, y) in zip(x_fed, y_fed):
  hash_table[hash_xy(x, y)] += 1
  - counts = hash_table.values()
  + counts = list(hash_table.values())
  unq = np.unique(counts)
  + print(unq)
  +
  assert len(unq) == 1, "Duplicate examples found"
  assert unq[0] == 1, "Federated example not found in original keras dataset"
  