fix formatting issues

Author: virajvchaudhari

test/integ_tests/test_all_notebooks.py

@@ -1,7 +1,7 @@
 import logging
 import os
-from importlib.machinery import SourceFileLoader
 import re
+from importlib.machinery import SourceFileLoader
 
 import pytest
 from jupyter_client import kernelspec
@@ -164,7 +164,7 @@ def check_cells_for_error_output(cells):
                 if output["output_type"] == "error":
                     pytest.fail("Found error output in cell: " + str(output))
 
-                     
+
 def execute_with_mocks(tb, mock_level, path_to_utils, path_to_mocks):
     # We do this check to make sure that the notebook can be executed (with mocks).
     tb.inject(
@@ -178,7 +178,7 @@ def execute_with_mocks(tb, mock_level, path_to_utils, path_to_mocks):
         run=False,
         before=0,
     )
-    
+
     # Uncomment all test sections in the notebook
     for i, cell in enumerate(tb.cells):
         if cell.get("cell_type") == "code" and "source" in cell:
@@ -187,39 +187,37 @@ def execute_with_mocks(tb, mock_level, path_to_utils, path_to_mocks):
                 # Uncomment the test section
                 modified_source = uncomment_test_section(source)
                 tb.cells[i]["source"] = modified_source
-    
+
     # Execute the notebook with the uncommented test sections
     tb.execute()
     test_mocks = SourceFileLoader("notebook_mocks", path_to_mocks).load_module()
     test_mocks.post_run(tb)
 
-    
 
 def uncomment_test_section(source):
     """Uncomment sections marked with # UNCOMMENT_TO_RUN."""
-    descriptive_comment = re.compile(r'^\s*##\s')
-    regular_comment = re.compile(r'^\s*#\s?')
-    
-    
+    descriptive_comment = re.compile(r"^\s*##\s")
+    regular_comment = re.compile(r"^\s*#\s?")
+
     lines = source.splitlines()
     result = []
     uncomment_mode = False
-    
+
     for line in lines:
         stripped_line = line.strip()
-        
+
         if stripped_line == "# UNCOMMENT_FOR_TEST":
             uncomment_mode = True
             result.append(line)
             continue
-            
+
         if uncomment_mode:
             if descriptive_comment.match(stripped_line):
                 # Handle descriptive comments (##)
                 result.append(f"# {stripped_line[3:].lstrip()}")
             elif regular_comment.match(stripped_line):
                 # Handle regular comments (#)
-                result.append(regular_comment.sub('', line))
+                result.append(regular_comment.sub("", line))
             elif stripped_line:
                 # Non-empty, non-comment line
                 uncomment_mode = False
@@ -229,5 +227,5 @@ def uncomment_test_section(source):
                 result.append(line)
         else:
             result.append(line)
-            
-    return '\n'.join(result)
+
+    return "\n".join(result)