Test if pedantic works for codspeed

egraphs-good · Feb 27, 2025 · 95ed3e8 · 95ed3e8
1 parent 3489a6b
commit 95ed3e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tests/test_array_api.py b/python/tests/test_array_api.py
@@ -412,7 +412,7 @@ def test_optimize(self, snapshot_py, benchmark):
         y = NDArray.var("y")
         with egraph.set_current():
             expr = lda(X, y)
-            simplified = benchmark(simplify_lda, egraph, expr)
+            simplified = benchmark.pendatic(simplify_lda, args=(egraph, expr))
         assert str(simplified) == snapshot_py
 
     # @pytest.mark.xfail(reason="Original source is not working")