From 32b0e2443fa8d8a409151753cf4b3e105a0dfb39 Mon Sep 17 00:00:00 2001 From: Jirka Date: Tue, 26 Sep 2023 10:34:14 +0200 Subject: [PATCH] _irs_mis_sz_fn --- tests/unittests/retrieval/helpers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unittests/retrieval/helpers.py b/tests/unittests/retrieval/helpers.py index 5b4cf9a24b6..66820708207 100644 --- 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