fix typing to fix pre-commit (#49)

pixee · Sep 27, 2023 · df11542 · df11542
1 parent ca2b7f5
commit df11542
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/codemodder/registry.py b/src/codemodder/registry.py
@@ -1,8 +1,8 @@
 from dataclasses import dataclass, asdict
+from importlib.abc import Traversable
 from importlib.resources import files
 from importlib.metadata import entry_points
 from typing import Optional
-from types import ModuleType
 
 from codemodder.logging import logger
 
@@ -23,8 +23,8 @@ class _CodemodWrapper:
 
     origin: str
     codemod: type
-    docs_module: ModuleType
-    semgrep_config_module: ModuleType
+    docs_module: Traversable
+    semgrep_config_module: Traversable
 
     def __call__(self, *args, **kwargs):
         return self.codemod(*args, **kwargs)