Fix periodically failing test

jloveric · May 19, 2024 · 7c80be0 · 7c80be0
1 parent 2dfee79
commit 7c80be0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -1,7 +1,5 @@
-
-<!---
 [![CI](https://github.com/jloveric/high-order-layers-torch/actions/workflows/python-app.yml/badge.svg)](https://github.com/jloveric/high-order-layers-torch/actions/workflows/python-app.yml)
---->
+
 
 # Piecewise Polynomial in PyTorch
 

diff --git a/tests/test_sparse_optimizer.py b/tests/test_sparse_optimizer.py
@@ -53,6 +53,6 @@ def test_sparse_lion():
     assert torch.any((w_after - w_before) > 0)
 
     out_final = layer(x)
-    diff = out_final - out
+    diff = torch.abs(out_final - out)
     assert torch.any(diff > 0)
     print("diff", diff)