prune-packages

.github/assistant.py

@@ -0,0 +1,22 @@
+from pprint import pprint
+
+import fire
+
+
+class AssistantCLI:
+    @staticmethod
+    def prune_packages(req_file: str, *pkgs: str) -> None:
+        """Prune packages from requirement file."""
+        with open(req_file) as fp:
+            lines = fp.readlines()
+
+        for pkg in pkgs:
+            lines = [ln for ln in lines if not ln.startswith(pkg)]
+        pprint(lines)
+
+        with open(req_file, "w") as fp:
+            fp.writelines(lines)
+
+
+if __name__ == "__main__":
+    fire.Fire(AssistantCLI)

.github/workflows/ci_test-conda.yml

@@ -20,8 +20,7 @@ jobs:
         python-version: ["3.8"]
         pytorch-version: ["1.4", "1.5", "1.6", "1.7", "1.8", "1.9", "1.10", "1.11"]
         include:
-          - python-version: 3.7
-            pytorch-version: '1.3'
+          - {python-version: 3.7, pytorch-version: '1.3'}
     env:
       PYTEST_ARTEFACT: test-conda-py${{ matrix.python-version }}-pt${{ matrix.pytorch-version }}.xml
       TRANSFORMERS_CACHE: .cache/huggingface/
@@ -83,7 +82,8 @@ jobs:
         conda list
         pip --version
         python ./requirements/adjust-versions.py requirements.txt
-        python ./.github/prune-packages.py requirements/image.txt torchvision
+        python ./.github/assistant.py prune-packages requirements/image.txt torchvision
+        python ./.github/assistant.py prune-packages requirements/detection.txt torchvision
         pip install -q "numpy==1.20.0"  # try to fix cocotools for PT 1.4 & 1.9
         pip install --requirement requirements.txt --quiet
         pip install --requirement requirements/devel.txt --quiet

requirements/test.txt

@@ -9,6 +9,7 @@ twine>=3.2
 mypy>=0.790
 phmdoctest>=1.1.1
 pre-commit>=1.0
+fire
 
 cloudpickle>=1.3
 scikit-learn>=0.24