Skip to content

Commit 05ca45a

Browse files
authored
Crawler: fix missing generic types on subclasses. (#2760)
## Changes This PR simply adds in some missing generic annotations for crawler implementations.
1 parent 729e9ae commit 05ca45a

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

src/databricks/labs/ucx/hive_metastore/table_size.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ class TableSize:
2222
size_in_bytes: int
2323

2424

25-
class TableSizeCrawler(CrawlerBase):
25+
class TableSizeCrawler(CrawlerBase[TableSize]):
2626
def __init__(self, backend: SqlBackend, schema, include_databases: list[str] | None = None):
2727
"""
2828
Initializes a TablesSizeCrawler instance.

src/databricks/labs/ucx/hive_metastore/tables.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -476,7 +476,7 @@ def _describe(self, catalog: str, database: str, table: str) -> Table | None:
476476
return None
477477

478478

479-
class FasterTableScanCrawler(CrawlerBase):
479+
class FasterTableScanCrawler(CrawlerBase[Table]):
480480
"""
481481
FasterTableScanCrawler is a specialized version of TablesCrawler that uses spark._jsparkSession to utilize
482482
faster scanning with Scala APIs.

src/databricks/labs/ucx/hive_metastore/udfs.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ def key(self) -> str:
3333
return f"{self.catalog}.{self.database}.{self.name}".lower()
3434

3535

36-
class UdfsCrawler(CrawlerBase):
36+
class UdfsCrawler(CrawlerBase[Udf]):
3737
def __init__(self, backend: SqlBackend, schema: str, include_databases: list[str] | None = None):
3838
"""
3939
Initializes a UdfsCrawler instance.

0 commit comments

Comments
 (0)