diff options
Diffstat (limited to 'tests/test_nn_rewrite_core_train.py')
-rw-r--r-- | tests/test_nn_rewrite_core_train.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/test_nn_rewrite_core_train.py b/tests/test_nn_rewrite_core_train.py index 4493671..b001a09 100644 --- a/tests/test_nn_rewrite_core_train.py +++ b/tests/test_nn_rewrite_core_train.py @@ -62,7 +62,7 @@ def check_train( train_params=train_params, ) assert len(result) == 2 - assert all(res >= 0.0 for res in result), f"Results out of bound: {result}" + assert all(res >= 0.0 for res in result[0]), f"Results out of bound: {result}" assert output_file.is_file() |