Fix bug in tests

Bitbol-Lab · Feb 1, 2024 · 03e7957 · 03e7957
1 parent aefa67b
commit 03e7957
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/nbs/sequence_similarity_ops.ipynb b/nbs/sequence_similarity_ops.ipynb
@@ -155,7 +155,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "show_doc(soft_reciprocal_best_hits)"
+    "show_doc(soft_best_hits)"
    ]
   },
   {
@@ -164,7 +164,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "show_doc(hard_reciprocal_best_hits)"
+    "show_doc(hard_best_hits)"
    ]
   },
   {
@@ -180,7 +180,7 @@
     "    x = torch.randn(20, 6)\n",
     "    similarities = torch.cdist(x, x, p=1)\n",
     "    similarities /= similarities.max()\n",
-    "    srbh = soft_reciprocal_best_hits(similarities, group_slices=group_slices)\n",
+    "    srbh = soft_best_hits(similarities, group_slices=group_slices, reciprocal=True)\n",
     "\n",
     "    assert srbh.shape == similarities.shape\n",
     "    assert torch.all(torch.logical_and(srbh <= 1, srbh >= 0))\n",