Automated pre-commit update

Author: devin-ai-integration[bot]

src/codegen/sdk/core/interfaces/importable.py

@@ -8,6 +8,7 @@
 from codegen.sdk.core.expressions.expression import Expression
 from codegen.sdk.core.interfaces.has_name import HasName
 from codegen.sdk.core.node_id_factory import NodeId
+from codegen.sdk.core.symbol import Symbol
 from codegen.sdk.enums import EdgeType
 from codegen.sdk.extensions.autocommit import commiter
 from codegen.sdk.extensions.sort import sort_editables
@@ -17,7 +18,6 @@
     from codegen.sdk.codebase.codebase_graph import CodebaseGraph
     from codegen.sdk.core.import_resolution import Import
     from codegen.sdk.core.interfaces.editable import Editable
-    from codegen.sdk.core.symbol import Symbol
 
 Parent = TypeVar("Parent", bound="Editable")
 
@@ -76,10 +76,10 @@ def get_dependencies_to_depth(self, depth: Optional[int] = None) -> list[Union["
     @noapidoc
     def get_dependencies(self, usage_types: UsageType) -> list[Union["Symbol", "Import"]]:
         """Internal method for getting dependencies by usage type.
-        
+
         This is kept for backward compatibility and internal use.
         New code should use the dependencies property or get_dependencies_to_depth method.
-        
+
         Opposite of `usages`
         """
         avoid = set(self.descendant_symbols)
@@ -138,24 +138,24 @@ def _remove_internal_edges(self, edge_type: EdgeType | None = None) -> None:
     @noapidoc
     def _get_all_dependencies(self, max_depth: Optional[int] = None) -> list[Union["Symbol", "Import"]]:
         """Gets all dependencies up to specified depth.
-        
+
         Internal implementation of depth-based dependency traversal.
         """
-        dependency_map: dict[Self, list[Union["Symbol", "Import"]]] = {}
+        dependency_map: dict[Self, list[Union[Symbol, Import]]] = {}
         current_depth = 1
-        
+
         def _collect_deps(symbol: Self, depth: int) -> None:
             if depth > max_depth if max_depth is not None else False:
                 return
-                
+
             direct_deps = symbol.get_dependencies(UsageType.DIRECT)
             if symbol not in dependency_map:
                 dependency_map[symbol] = direct_deps
-                
+
             for dep in direct_deps:
                 if isinstance(dep, Symbol):
                     _collect_deps(dep, depth + 1)
-        
+
         _collect_deps(self, current_depth)
         all_deps = []
         for deps in dependency_map.values():

tests/unit/codegen/sdk/core/interfaces/test_importable_dependencies.py

@@ -60,7 +60,7 @@ def __init__(self, value: Optional[str] = None):
 """
     with get_codebase_session(tmpdir=tmpdir, programming_language=ProgrammingLanguage.PYTHON, files={"test.py": py_code}) as G:
         my_class = G.get_class("MyClass")
-        
+
         # Test that imports are included in dependencies
         deps = my_class.get_dependencies_to_depth(depth=1)
         assert len(deps) == 2  # Optional and str
@@ -85,7 +85,7 @@ def __init__(self, value: Optional[str] = None):
 """
     with get_codebase_session(tmpdir=tmpdir, programming_language=ProgrammingLanguage.PYTHON, files={"test.py": py_code}) as G:
         my_class = G.get_class("MyClass")
-        
+
         # Test that get_dependencies with DIRECT usage type works
         direct_deps = my_class.get_dependencies(UsageType.DIRECT)
         assert len(direct_deps) == 2
@@ -114,7 +114,7 @@ class B(A):
 """
     with get_codebase_session(tmpdir=tmpdir, programming_language=ProgrammingLanguage.PYTHON, files={"test.py": py_code}) as G:
         b_class = G.get_class("B")
-        
+
         # Test that depth=0 returns empty list
         deps = b_class.get_dependencies_to_depth(depth=0)
         assert len(deps) == 0