Update test/profiler/test_memory_profiler.py

malfet · web-flow · commit b4cb7c29887f · 2025-05-13T17:01:04.000-07:00
diff --git a/test/profiler/test_memory_profiler.py b/test/profiler/test_memory_profiler.py
@@ -1495,7 +1495,7 @@ def id_for_testing(key):
             for _, action, (key, version), size in prof._memory_profile().timeline
             # We generally don't care about tiny allocations during memory
             # profiling and they add a lot of noise to the unit test.
-            if size > 1024 and isinstance(key, _memory_profiler.TensorKey) 
+            if size > 1024 and isinstance(key, _memory_profiler.TensorKey)
         ]
 
         self.assertExpectedInline(

Original file line number	Diff line number	Diff line change
`@@ -1495,7 +1495,7 @@ def id_for_testing(key):`
`1495`	`1495`	`for _, action, (key, version), size in prof._memory_profile().timeline`
`1496`	`1496`	`# We generally don't care about tiny allocations during memory`
`1497`	`1497`	`# profiling and they add a lot of noise to the unit test.`
`1498`		`- if size > 1024 and isinstance(key, _memory_profiler.TensorKey)`
	`1498`	`+ if size > 1024 and isinstance(key, _memory_profiler.TensorKey)`
`1499`	`1499`	`]`
`1500`	`1500`
`1501`	`1501`	`self.assertExpectedInline(`