fix viz with untracked graph_rewrite calls (#8298)

* fix viz with untracked graph_rewrite calls

* mark as green
This commit is contained in:
qazal
2024-12-17 23:37:53 +02:00
committed by GitHub
parent 5977a3d8a6
commit 673a76398a
2 changed files with 3 additions and 4 deletions

View File

@@ -114,7 +114,6 @@ class TestViz(unittest.TestCase):
self.assertIs(ret[0], a.sqrt().sin()) # only rewrite
# NOTE: calling graph_rewrite when the function isn't decorated with track_rewrites should not VIZ
@unittest.expectedFailure
def test_rewrite_without_context(self):
def untracked_graph_rewrite(sink): return graph_rewrite(sink, symbolic)
@track_rewrites(named=True)