Skip to content

Commit 4d27f80

Browse files
committed
Add missing type hints to UsedTablesCrawler
1 parent 28e8934 commit 4d27f80

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/databricks/labs/ucx/source_code/used_table.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,14 +16,14 @@
1616
class UsedTablesCrawler(CrawlerBase[UsedTable]):
1717

1818
@classmethod
19-
def for_paths(cls, backend: SqlBackend, schema) -> UsedTablesCrawler:
19+
def for_paths(cls, backend: SqlBackend, schema: str) -> UsedTablesCrawler:
2020
return UsedTablesCrawler(backend, schema, "used_tables_in_paths")
2121

2222
@classmethod
23-
def for_queries(cls, backend: SqlBackend, schema) -> UsedTablesCrawler:
23+
def for_queries(cls, backend: SqlBackend, schema: str) -> UsedTablesCrawler:
2424
return UsedTablesCrawler(backend, schema, "used_tables_in_queries")
2525

26-
def __init__(self, backend: SqlBackend, schema: str, table: str):
26+
def __init__(self, backend: SqlBackend, schema: str, table: str) -> None:
2727
"""
2828
Initializes a DFSACrawler instance.
2929
@@ -33,7 +33,7 @@ def __init__(self, backend: SqlBackend, schema: str, table: str):
3333
"""
3434
super().__init__(backend=backend, catalog="hive_metastore", schema=schema, table=table, klass=UsedTable)
3535

36-
def dump_all(self, tables: Sequence[UsedTable]):
36+
def dump_all(self, tables: Sequence[UsedTable]) -> None:
3737
"""This crawler doesn't follow the pull model because the fetcher fetches data for 3 crawlers, not just one
3838
It's not **bad** because all records are pushed at once.
3939
Providing a multi-entity crawler is out-of-scope of this PR

0 commit comments

Comments
 (0)