Merge pull request #919 from adamjstewart/deps/mock

.github/workflows/pypi.yml

@@ -16,7 +16,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip uv
-        python -m uv pip install setuptools wheel twine mock
+        python -m uv pip install setuptools wheel twine
     - name: Build and publish
       env:
         TWINE_USERNAME: ${{ secrets.PYPI_USERNAME }}

docker/Dockerfile.dev

@@ -4,7 +4,7 @@ WORKDIR /tmp/smp/
 
 COPY ./requirements.txt /tmp/smp/requirements.txt
 RUN pip install -r requirements.txt
-RUN pip install pytest mock
+RUN pip install pytest
 
 COPY . /tmp/smp/
 RUN pip install .

pyproject.toml

@@ -39,7 +39,6 @@ docs = [
     'sphinx-book-theme==1.1.2',
 ]
 test = [
-    'mock',
     'pytest',
     'ruff==0.5.2',
 ]

tests/test_models.py

@@ -1,10 +1,6 @@
-import sys
-import mock
 import pytest
 import torch
 
-# mock detection module
-sys.modules["torchvision._C"] = mock.Mock()
 import segmentation_models_pytorch as smp  # noqa
 
 

tests/test_preprocessing.py

@@ -1,9 +1,5 @@
-import sys
-import mock
 import numpy as np
 
-# mock detection module
-sys.modules["torchvision._C"] = mock.Mock()
 import segmentation_models_pytorch as smp  # noqa