commit b73acb198e5419f08c12c893b1a41bd847b41860
author Anthony Sottile <asottile@umich.edu> Sat Apr 06 14:35:41 2024 -0400
committer GitHub <noreply@github.com> Sat Apr 06 14:35:41 2024 -0400
tree 63277c8ef13920ad051bb2e289c5bacd6adbbf05
parent 2c9f875913ee60ca25ce70243dc24d5b6415598c
parent eadcce2e03893cb90c5da35286855e7867bef0f7

Merge pull request #850 from ericfrederich/fix-requirements-fixer

Stop pkg_resources from being added as well

pre_commit_hooks/requirements_txt_fixer.py

diff --git a/pre_commit_hooks/requirements_txt_fixer.py b/pre_commit_hooks/requirements_txt_fixer.py
index 261acc9..07b57e1 100644
--- a/pre_commit_hooks/requirements_txt_fixer.py
+++ b/pre_commit_hooks/requirements_txt_fixer.py
@@ -115,7 +115,10 @@
     # which is automatically added by broken pip package under Debian
     requirements = [
         req for req in requirements
-        if req.value != b'pkg-resources==0.0.0\n'
+        if req.value not in [
+            b'pkg-resources==0.0.0\n',
+            b'pkg_resources==0.0.0\n',
+        ]
     ]
 
     # sort the requirements and remove duplicates

tests/requirements_txt_fixer_test.py

diff --git a/tests/requirements_txt_fixer_test.py b/tests/requirements_txt_fixer_test.py
index c400be1..c0d2c65 100644
--- a/tests/requirements_txt_fixer_test.py
+++ b/tests/requirements_txt_fixer_test.py
@@ -82,6 +82,8 @@
         ),
         (b'bar\npkg-resources==0.0.0\nfoo\n', FAIL, b'bar\nfoo\n'),
         (b'foo\npkg-resources==0.0.0\nbar\n', FAIL, b'bar\nfoo\n'),
+        (b'bar\npkg_resources==0.0.0\nfoo\n', FAIL, b'bar\nfoo\n'),
+        (b'foo\npkg_resources==0.0.0\nbar\n', FAIL, b'bar\nfoo\n'),
         (
             b'git+ssh://git_url@tag#egg=ocflib\nDjango\nijk\n',
             FAIL,