linting

funkelab · Nov 10, 2024 · f5d49b7 · f5d49b7
1 parent 205f4d6
commit f5d49b7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -25,3 +25,6 @@ repos:
     rev: v1.13.0
     hooks:
       - id: mypy
+        additional_dependencies:
+          - cython
+          - setuptools
diff --git a/src/witty/compile_module.py b/src/witty/compile_module.py
@@ -12,16 +12,12 @@
 from Cython.Build import cythonize
 from Cython.Build.Inline import build_ext
 from Cython.Utils import get_cython_cache_dir
+from setuptools import Distribution, Extension
 
 if TYPE_CHECKING:
     from collections.abc import Iterator, Sequence
     from types import ModuleType
 
-try:
-    from distutils.core import Distribution, Extension  # until Python 3.11
-except ImportError:
-    from setuptools import Distribution, Extension
-
 
 def compile_module(
     source_pyx: str,