fix pre_gen_project error

beam-pyio · Jun 27, 2024 · ea5b73f · ea5b73f
1 parent ca5d358
commit ea5b73f
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/hooks/pre_gen_project.py b/hooks/pre_gen_project.py
@@ -1,16 +1,16 @@
 import sys
 
-from pkg_resources import (
-    get_distribution,
-    packaging,
-)
-
 MIN_CC_VERSION = "2.0.0"
 
+try:
+    import cookiecutter # type: ignore
+except ImportError:
+    print(f"ERROR: please install cookiecutter >= {MIN_CC_VERSION}")
+    sys.exit(1)
+
+cc_version = cookiecutter.__version__
 # assert cookiecutter >= 2.0.0
-cc_version = packaging.version.parse(get_distribution("cookiecutter").version)
-min_version = packaging.version.parse(MIN_CC_VERSION)
-if cc_version < min_version:
+if cc_version < MIN_CC_VERSION:
     print(
         f"ERROR: please install cookiecutter >= {MIN_CC_VERSION} (current "
         f"version is {cc_version}):\n"