43a60c9f4380448b9013efa4fb021796e7e1ff3c,test/test_mahalanobis_mixin.py,,test_components_is_2D,#Any#Any#,281

Before Change

  trunc_data = input_data[..., :1]
  // we drop duplicates that might have been formed, i.e. of the form
  // aabc or abcc or aabb for quadruplets, and aa for pairs.
  if isinstance(estimator, _QuadrupletsClassifierMixin):
    for slice_idx in [slice(0, 2), slice(2, 4)]:
      pairs = trunc_data[:, slice_idx, :]
      diffs = pairs[:, 1, :] - pairs[:, 0, :]
      to_keep = np.where(np.abs(diffs.ravel()) > 1e-9)
      trunc_data = trunc_data[to_keep]
      labels = labels[to_keep]
  elif isinstance(estimator, _PairsClassifierMixin):
    diffs = trunc_data[:, 1, :] - trunc_data[:, 0, :]
    to_keep = np.where(np.abs(diffs.ravel()) > 1e-9)
    trunc_data = trunc_data[to_keep]
    labels = labels[to_keep]
  model.fit(*remove_y_quadruplets(estimator, trunc_data, labels))
  assert model.components_.shape == (1, 1)  // the components must be 2D

After Change

  elif isinstance(estimator, _PairsClassifierMixin):
    pairs_idx = [[0, 1]]
  else:
    pairs_idx = []

  for pair_idx in pairs_idx:
    pairs = trunc_data[:, pair_idx, :]
    diffs = pairs[:, 1, :] - pairs[:, 0, :]

Instances