... intermediate progress

databrickslabs · May 3, 2024 · 16502d7 · 16502d7
1 parent 8ba3e87
commit 16502d7
Show file tree

Hide file tree

Showing 5 changed files with 1 addition and 8 deletions.
diff --git a/src/databricks/labs/ucx/source_code/files.py b/src/databricks/labs/ucx/source_code/files.py
@@ -3,8 +3,6 @@
 import ast
 import logging
 from pathlib import Path
-from collections.abc import Callable
-from collections.abc import Iterable
 
 from databricks.labs.ucx.source_code.path_lookup import PathLookup
 from databricks.sdk.service.workspace import Language

diff --git a/src/databricks/labs/ucx/source_code/graph.py b/src/databricks/labs/ucx/source_code/graph.py
@@ -59,7 +59,7 @@ def register_dependency(self, dependency: Dependency) -> MaybeGraph:
         if not container:
             problem = DependencyProblem('dependency-register-failed', 'Failed to register dependency', dependency.path)
             return MaybeGraph(child_graph, [problem])
-        problems = container.build_dependency_graph(child_graph)
+        problems = container.build_dependency_graph(child_graph, self._path_lookup)
         return MaybeGraph(child_graph, problems)
 
     def locate_dependency(self, path: Path) -> MaybeGraph:

diff --git a/src/databricks/labs/ucx/source_code/site_packages.py b/src/databricks/labs/ucx/source_code/site_packages.py
@@ -1,12 +1,9 @@
 from __future__ import annotations
 
 import os
-from dataclasses import dataclass
 from pathlib import Path
-from collections.abc import Callable
 from databricks.labs.ucx.source_code.files import FileLoader
 from databricks.labs.ucx.source_code.path_lookup import PathLookup
-from databricks.labs.ucx.source_code.files import FileLoader, SysPathProvider
 from databricks.labs.ucx.source_code.graph import (
     Dependency,
     WrappingLoader,

diff --git a/tests/unit/source_code/test_notebook.py b/tests/unit/source_code/test_notebook.py
@@ -3,7 +3,6 @@
 import re
 
 import pytest
-from databricks.labs.ucx.source_code.path_lookup import PathLookup
 from databricks.sdk.service.workspace import Language, ObjectType, ObjectInfo
 from databricks.sdk import WorkspaceClient
 

diff --git a/tests/unit/source_code/test_s3fs.py b/tests/unit/source_code/test_s3fs.py
@@ -9,7 +9,6 @@
     DependencyGraphBuilder,
 )
 from databricks.labs.ucx.source_code.files import FileLoader, LocalFileResolver
-from databricks.labs.ucx.source_code.path_lookup import PathLookup
 from databricks.labs.ucx.source_code.whitelist import WhitelistResolver, Whitelist
 from tests.unit import (
     _load_sources,