lint

pytorch · Jun 18, 2024 · e651719 · e651719
1 parent 04a4387
commit e651719
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/test_cuda.py b/test/test_cuda.py
@@ -2319,7 +2319,9 @@ def test_graph_warn_if_has_zero_nodes(self):
     @unittest.skipIf(
         not TEST_CUDA_GRAPH, "CUDA >= 11.0 or ROCM >= 5.3 required for graphs"
     )
-    @unittest.skipIf(IS_JETSON, "oom reporting has issues on jetson igx due to partial nvml support")
+    @unittest.skipIf(
+        IS_JETSON, "oom reporting has issues on jetson igx due to partial nvml support"
+    )
     def test_graph_capture_oom(self):
         oom_regex = (
             "would exceed allowed memory" if TEST_CUDAMALLOCASYNC else "out of memory"