lint: fix custom mypy cache dir setting

fjahr · Sep 20, 2023 · b5825de · b5825de
1 parent 99ce836
commit b5825de
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test/lint/lint-python.py b/test/lint/lint-python.py
@@ -9,14 +9,18 @@
 """
 
 import os
+from pathlib import Path
 import subprocess
 import sys
 
 from importlib.metadata import metadata, PackageNotFoundError
 
+# Customize mypy cache dir via environment variable
+base_dir = Path(__file__).parent.parent.parent
+cache_dir = base_dir / "test" / ".mypy_cache"
+os.environ["MYPY_CACHE_DIR"] = str(cache_dir)
 
 DEPS = ['flake8', 'lief', 'mypy', 'pyzmq']
-MYPY_CACHE_DIR = f"{os.getenv('BASE_ROOT_DIR', '')}/test/.mypy_cache"
 
 # All .py files, except those in src/ (to exclude subtrees there)
 FLAKE_FILES_ARGS = ['git', 'ls-files', '*.py', ':!:src/*.py']