pre-commit cleanup

Author: KRRT7

codeflash/discovery/discover_unit_tests.py

@@ -167,31 +167,34 @@ def visit_ImportFrom(self, node: ast.ImportFrom) -> None:
             if alias.name == "*":
                 # Star imports - we can't know what's imported, so be conservative
                 self.imported_names.add("*")
-            else:
-                imported_name = alias.asname if alias.asname else alias.name
+            elif imported_name := (alias.asname if alias.asname else alias.name):
                 self.imported_names.add(imported_name)
-
-                # Check if this import matches any target function
                 if alias.name in self.function_names_to_find:
                     self.found_target_functions.add(alias.name)
         self.generic_visit(node)
 
     def visit_Call(self, node: ast.Call) -> None:
         """Handle dynamic imports like importlib.import_module() or __import__()."""
-        if isinstance(node.func, ast.Name) and node.func.id == "__import__" and node.args:
+        if (
+            isinstance(node.func, ast.Name)
+            and node.func.id == "__import__"
+            and node.args
+            and isinstance(node.args[0], ast.Constant)
+            and isinstance(node.args[0].value, str)
+        ):
             # __import__("module_name")
-            if isinstance(node.args[0], ast.Constant) and isinstance(node.args[0].value, str):
-                self.imported_modules.add(node.args[0].value)
+            self.imported_modules.add(node.args[0].value)
         elif (
             isinstance(node.func, ast.Attribute)
             and isinstance(node.func.value, ast.Name)
             and node.func.value.id == "importlib"
             and node.func.attr == "import_module"
             and node.args
+            and isinstance(node.args[0], ast.Constant)
+            and isinstance(node.args[0].value, str)
         ):
             # importlib.import_module("module_name")
-            if isinstance(node.args[0], ast.Constant) and isinstance(node.args[0].value, str):
-                self.imported_modules.add(node.args[0].value)
+            self.imported_modules.add(node.args[0].value)
         self.generic_visit(node)
 
     def visit_Name(self, node: ast.Name) -> None: