chore(checks): fix type hints

WeblateOrg · May 6, 2024 · f45158e · f45158e
1 parent 0e386e6
commit f45158e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/weblate/checks/base.py b/weblate/checks/base.py
@@ -22,6 +22,8 @@
 
     from django_stubs_ext import StrOrPromise
 
+    from weblate.trans.models import Unit
+
 
 class Check:
     """Basic class for checks."""
@@ -202,7 +204,7 @@ def handle_batch(self, unit, component):
         component.batched_checks.add(self.check_id)
         return self.check_id in unit.all_checks_names
 
-    def check_component(self, component) -> bool:
+    def check_component(self, component) -> Iterable[Unit]:
         raise NotImplementedError
 
     def perform_batch(self, component) -> None:

diff --git a/weblate/checks/format.py b/weblate/checks/format.py
@@ -266,7 +266,7 @@ class BaseFormatCheck(TargetCheck):
     regexp: Pattern[str]
     plural_parameter_regexp: Pattern[str] | None = None
     default_disabled = True
-    normalize_remove = None
+    normalize_remove: str | None = None
 
     def check_target_unit(self, sources, targets, unit):
         """Check single unit, handling plurals."""