Add absolute tolerance parameter to assert_allclose

njzjz · Nov 29, 2024 · dae05d8 · dae05d8
1 parent f8986de
commit dae05d8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/source/tests/universal/common/cases/model/utils.py b/source/tests/universal/common/cases/model/utils.py
@@ -189,6 +189,7 @@ def test_forward(self) -> None:
                     ret[0][kk][0],
                     ret[0][kk][1],
                     err_msg=f"compare {kk} between frame 0 and 1",
+                    atol=aprec,
                 )
 
             subret = []