aboutsummaryrefslogtreecommitdiff
path: root/tests/test_nn_rewrite_core_train.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_nn_rewrite_core_train.py')
-rw-r--r--tests/test_nn_rewrite_core_train.py2
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()