Просмотр исходного кода

Add type annotations to dulwich/merge_drivers.py

Jelmer Vernooij 5 месяцев назад
Родитель
Сommit
6127c9bfba
1 измененных файлов с 4 добавлено и 4 удалено
  1. 4 4
      dulwich/merge_drivers.py

+ 4 - 4
dulwich/merge_drivers.py

@@ -23,7 +23,7 @@
 import os
 import os
 import subprocess
 import subprocess
 import tempfile
 import tempfile
-from typing import Any, Optional, Protocol
+from typing import Any, Callable, Optional, Protocol
 
 
 from .config import Config
 from .config import Config
 
 
@@ -149,12 +149,12 @@ class MergeDriverRegistry:
         # Register built-in drivers
         # Register built-in drivers
         self._register_builtin_drivers()
         self._register_builtin_drivers()
 
 
-    def _register_builtin_drivers(self):
+    def _register_builtin_drivers(self) -> None:
         """Register built-in merge drivers."""
         """Register built-in merge drivers."""
         # The "text" driver is the default three-way merge
         # The "text" driver is the default three-way merge
         # We don't register it here as it's handled by the default merge code
         # We don't register it here as it's handled by the default merge code
 
 
-    def register_driver(self, name: str, driver: MergeDriver):
+    def register_driver(self, name: str, driver: MergeDriver) -> None:
         """Register a merge driver instance.
         """Register a merge driver instance.
 
 
         Args:
         Args:
@@ -163,7 +163,7 @@ class MergeDriverRegistry:
         """
         """
         self._drivers[name] = driver
         self._drivers[name] = driver
 
 
-    def register_factory(self, name: str, factory):
+    def register_factory(self, name: str, factory: Callable[[], MergeDriver]) -> None:
         """Register a factory function for creating merge drivers.
         """Register a factory function for creating merge drivers.
 
 
         Args:
         Args: