code: def test_approx_min_k(self, qy_shape, db_shape, dtype, k, recall):
      rng = jtu.rand_default(self.rng())
      qy = rng(qy_shape, dtype)
      db = rng(db_shape, dtype)
      scores = lax.dot(qy, db)
      _, gt_args = lax.top_k(-scores, k)
      _, ann_args = lax.approx_min_k(scores, k, recall_target=recall)
      ann_recall = compute_recall(
          np.asarray(ann_args), np.asarray(gt_args)
      )
      self.assertGreaterEqual(ann_recall, recall * 0.9)
  
file_path: jax/tests/ann_test.py
imports: from functools import partial
  import math
  from absl.testing import absltest
  import numpy
  import jax
  from jax import lax
  from jax._src import test_util
  from jax import config
label_desc: Use image node_id 2 for calling the AnnTest obj's underlying member method code with example usage: obj.test_approx_min_k(qy_shape, db_shape, dtype, k, recall) without return types