fix linter errors

Author: mohamedashrraf222

codeflash/api/cfapi.py

@@ -131,17 +131,21 @@ def validate_api_key(api_key: Optional[str] = None) -> None:
     # First check if API key exists
     if not api_key:
         api_key = get_codeflash_api_key()
-    
+
+    def raise_os_error(msg: str) -> None:
+        """Raise OSError with the given message."""
+        raise OSError(msg)
+
     # Make the request to validate the API key
     try:
         response = make_cfapi_request(
-            endpoint="/cli-get-user", 
-            method="GET", 
-            extra_headers={"cli_version": __version__}, 
+            endpoint="/cli-get-user",
+            method="GET",
+            extra_headers={"cli_version": __version__},
             api_key=api_key,
-            suppress_errors=True  # Don't log errors yet, we'll handle it below
+            suppress_errors=True,  # Don't log errors yet, we'll handle it below
         )
-        
+
         if response.status_code == 403:
             msg = (
                 "Invalid Codeflash API key. The API key you provided is not valid.\n"
@@ -150,27 +154,27 @@ def validate_api_key(api_key: Optional[str] = None) -> None:
                 "For more information, refer to the documentation at "
                 "https://docs.codeflash.ai/getting-started/codeflash-github-actions#add-your-api-key-to-your-repository-secrets."
             )
-            raise OSError(msg)
+            raise_os_error(msg)
         elif response.status_code != 200:
             msg = (
                 f"Failed to validate API key with Codeflash API (status {response.status_code}).\n"
                 "Please verify your API key is correct.\n"
                 "You can generate a new one at https://app.codeflash.ai/app/apikeys"
             )
-            raise OSError(msg)
-            
+            raise_os_error(msg)
+
         # Check for version updates
         if response.status_code == 200:
             try:
                 resp_json = response.json()
                 min_version = resp_json.get("min_version")
                 if min_version and version.parse(min_version) > version.parse(__version__):
                     msg = "Your Codeflash CLI version is outdated. Please update to the latest version using `pip install --upgrade codeflash`."
-                    raise OSError(msg)
+                    raise_os_error(msg)
             except (json.JSONDecodeError, KeyError, TypeError):
                 # If response is not JSON or doesn't have min_version, that's okay
                 pass
-                
+
     except OSError:
         # Re-raise OSError as-is
         raise

codeflash/optimization/optimizer.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import ast
+import contextlib
 import copy
 import os
 import tempfile
@@ -507,10 +508,8 @@ def run_with_args(args: Namespace) -> None:
         # Log the error if we have an optimizer instance with cleanup
         if optimizer:
             logger.error(f"Error during optimization: {e}")
-            try:
+            with contextlib.suppress(Exception):
                 optimizer.cleanup_temporary_paths()
-            except Exception:
-                pass
         else:
             logger.error(f"Error initializing optimizer: {e}")
         raise