diff --git a/tests/test_cnn.py b/tests/test_cnn.py index 070ec35..a074d5e 100644 --- a/tests/test_cnn.py +++ b/tests/test_cnn.py @@ -8,6 +8,7 @@ import torch from PIL import Image import pytest +import warnings from torchvision.transforms import transforms from imagededup.methods.cnn import CNN @@ -891,7 +892,8 @@ def test_find_duplicates_encoding_integration(cnn): } encodings = cnn.encode_images(TEST_IMAGE_DIR_MIXED) - with pytest.warns(None): + with warnings.catch_warnings(): + warnings.simplefilter("ignore") duplicates = cnn.find_duplicates( encoding_map=encodings, min_similarity_threshold=0.9, scores=True, outfile=False ) diff --git a/tests/test_hashing.py b/tests/test_hashing.py index d7409ca..3e38210 100644 --- a/tests/test_hashing.py +++ b/tests/test_hashing.py @@ -5,6 +5,7 @@ from multiprocessing import cpu_count import pytest +import warnings import numpy as np from imagededup.methods.hashing import Hashing, PHash, DHash, AHash, WHash @@ -800,7 +801,8 @@ def test_find_duplicates_encoding_map_input(): 'ukbench09268.jpg': 'c73c36c2da2f29c9', } phasher = PHash() - with pytest.warns(None): + with warnings.catch_warnings(): + warnings.simplefilter("ignore") duplicate_dict = phasher.find_duplicates( encoding_map=encoding, max_distance_threshold=10 )