Update code_context_extractor.py

Author: KRRT7

codeflash/context/code_context_extractor.py

@@ -491,16 +491,14 @@ def parse_code_and_prune_cst(
             module, target_functions, helpers_of_helper_functions, remove_docstrings=remove_docstrings
         )
     else:
-        msg = f"Unknown code_context_type: {code_context_type}"
-        raise TypeError(msg)
+        raise TypeError(f"Unknown code_context_type: {code_context_type}")  # noqa: EM102
 
     if not found_target:
         raise ValueError("No target functions found in the provided code")
     if filtered_node and isinstance(filtered_node, cst.Module):
         return str(filtered_node.code)
     return ""
 
-
 def prune_cst_for_read_writable_code(  # noqa: PLR0911
     node: cst.CSTNode, target_functions: set[str], prefix: str = ""
 ) -> tuple[cst.CSTNode | None, bool]: