From da61b40604dadae662fecd7078308ee49a3342f9 Mon Sep 17 00:00:00 2001
From: qazal <77887910+Qazalin@users.noreply.github.com>
Date: Thu, 4 Sep 2025 23:59:32 +0300
Subject: [PATCH] some viz tests don't need track_rewrites (#12010)

---
 test/unit/test_viz.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/test/unit/test_viz.py b/test/unit/test_viz.py
index f1e3c10b82..23cf06489c 100644
--- a/test/unit/test_viz.py
+++ b/test/unit/test_viz.py
@@ -70,7 +70,6 @@ class TestViz(BaseTestViz):
 
   def test_rewrite_location(self):
     def inner(sink): return graph_rewrite(sink, PatternMatcher([]))
-    @track_rewrites(name=True)
     def outer(sink): return inner(sink)
     outer(UOp.variable("a", 1, 10))
     lst = get_viz_list()
@@ -181,7 +180,6 @@ class TestVizTree(BaseTestViz):
     c = UOp.variable("c",0,10)
     d = UOp.variable("d",0,10)
     sink = UOp.sink(a+b, c+d)
-    @track_rewrites()
     def tree_rewrite(): return graph_rewrite(sink, root, name="root")
     tree_rewrite()
     lst = get_viz_list()
@@ -256,7 +254,7 @@ class TestVizIntegration(BaseTestViz):
     self.assertEqual(len([n for n in graph.values() if repr(metadata) in n["label"]]), 1)
 
   # tracing also works without a track_rewrites context
-  # all graph_rewrites get put into the a default group
+  # all graph_rewrites get put into the default group
   def test_default_tracing(self):
     def test(root):
       return graph_rewrite(root, sym)