diff --git a/pyproject.toml b/pyproject.toml index 99b54249b..4b8666ceb 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -318,7 +318,7 @@ dev = [ "sybil==6.1.1", "tenacity==8.3.0", "torch==2.3.0", - "types-docker==7.0.0.20240513", + "types-docker==7.0.0.20240515", "types-Pillow==10.2.0.20240511", "types-PyYAML==6.0.12.20240311", "types-requests==2.31.0.20240406", diff --git a/tests/mock_vws/test_docker.py b/tests/mock_vws/test_docker.py index 7c8d68d6b..7864c9069 100644 --- a/tests/mock_vws/test_docker.py +++ b/tests/mock_vws/test_docker.py @@ -13,7 +13,6 @@ import requests from docker.errors import BuildError, NotFound from docker.models.containers import Container -from docker.models.images import Image from docker.models.networks import Network from mock_vws.database import VuforiaDatabase from tenacity import retry @@ -26,6 +25,8 @@ import io from collections.abc import Iterator + from docker.models.images import Image + # We do not cover this function because hitting particular branches depends on # timing. @@ -91,7 +92,7 @@ def fixture_custom_bridge_network() -> Iterator[Network]: network.disconnect(container=container) # pyright: ignore[reportUnknownMemberType] container.stop() # pyright: ignore[reportUnknownMemberType] container.remove(v=True, force=True) # pyright: ignore[reportUnknownMemberType] - assert isinstance(container.image, Image) # pyright: ignore[reportUnknownMemberType] + assert container.image is not None images_to_remove.add(container.image) # This does leave behind untagged images.