_irs_mis_sz_fn

Lightning-AI · Sep 26, 2023 · 32b0e24 · 32b0e24
1 parent 43b4d3a
commit 32b0e24
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unittests/retrieval/helpers.py b/tests/unittests/retrieval/helpers.py
@@ -136,7 +136,7 @@ def _concat_tests(*tests: Tuple[Dict]) -> Dict:
     "argnames": "preds,target,message,metric_args",
     "argvalues": [
         # check input shapes are consistent (func)
-        (_irs_mis_sz_fn.preds, _irs_bad_sz_fn.target, "`preds` and `target` must be of the same shape", {}),
+        (_irs_bad_sz_fn.preds, _irs_bad_sz_fn.target, "`preds` and `target` must be of the same shape", {}),
         # check input tensors are not empty
         (_irs_empty.preds, _irs_empty.target, "`preds` and `target` must be non-empty and non-scalar tensors", {}),
         # check on input dtypes
@@ -150,7 +150,7 @@ def _concat_tests(*tests: Tuple[Dict]) -> Dict:
     "argnames": "preds,target,message,metric_args",
     "argvalues": [
         # check input shapes are consistent (func)
-        (_irs_mis_sz_fn.preds, _irs_bad_sz_fn.target, "`preds` and `target` must be of the same shape", {}),
+        (_irs_bad_sz_fn.preds, _irs_bad_sz_fn.target, "`preds` and `target` must be of the same shape", {}),
         # check input tensors are not empty
         (_irs_empty.preds, _irs_empty.target, "`preds` and `target` must be non-empty and non-scalar tensors", {}),
         # check on input dtypes