fix for biencoder embedding

webis-de · Dec 9, 2024 · 56f2f08 · 56f2f08
1 parent 91eadd3
commit 56f2f08
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_loss.py b/tests/test_loss.py
@@ -125,7 +125,9 @@ def test_in_batch_loss_func(loss_func: InBatchLossFunction, output: LightningIRO
 def test_regularization_loss_func(loss_func: RegularizationLossFunction, embeddings: torch.Tensor):
     loss = loss_func.compute_loss(
         BiEncoderOutput(
-            None, BiEncoderEmbedding(embeddings, torch.empty(0)), BiEncoderEmbedding(embeddings, torch.empty(0))
+            None,
+            BiEncoderEmbedding(embeddings, torch.empty(0), None),
+            BiEncoderEmbedding(embeddings, torch.empty(0), None),
         )
     )
     assert loss >= 0