diff --git a/fastvideo/utils/collect_env.py b/collect_env.py
similarity index 99%
rename from fastvideo/utils/collect_env.py
rename to collect_env.py
index 1458a8468..44082ff2f 100644
--- a/fastvideo/utils/collect_env.py
+++ b/collect_env.py
@@ -81,6 +81,7 @@ DEFAULT_CONDA_PATTERNS = {
 DEFAULT_PIP_PATTERNS = {
     "torch",
     "numpy",
+    "mypy",
     "flake8",
     "triton",
     "optree",