Skip to content

Commit

Permalink
Merge pull request #1 from leferrad/DAI-Labgh-88-update-dependencies
Browse files Browse the repository at this point in the history
Gh 88 update dependencies
  • Loading branch information
leferrad authored Apr 4, 2023
2 parents 513912f + 469fc96 commit 3baf252
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 5 deletions.
1 change: 0 additions & 1 deletion setup.cfg
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,6 @@ exclude = docs, .tox, .git, __pycache__, .ipynb_checkpoints
ignore = # keep empty to prevent default ignores

[isort]
include_trailing_comment = True
line_length = 99
lines_between_types = 0
multi_line_output = 4
Expand Down
4 changes: 2 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,9 +14,9 @@
install_requires = [
'imageio>=2.4.1,<2.5.0',
'reedsolo==0.3',
'scipy>=1.1.0,<1.2.0',
'scipy>=1.1.0',
'tqdm>=4.28.1',
'numpy>=1.15.4<1.16.0',
'numpy>=1.15.4',
'Pillow>=5.0.0,<8.0.0',
'torch>=1.0.0,<2.0.0',
'torchvision',
Expand Down
8 changes: 6 additions & 2 deletions tests/test_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -981,7 +981,9 @@ def test_decode_image_zero_candidates(self, mock_bits, mock_imread, mock_float_t

steganogan.decoder.return_value.view.return_value.__gt__.return_value = _image

bits_image = _image.data.cpu.return_value.numpy.return_value.tolist.return_value
bits_image = (
_image.data.int.return_value.cpu.return_value.numpy.return_value.tolist.return_value
)

mock_bytearray_to_text.return_value = None
mock_os.return_value = True
Expand Down Expand Up @@ -1029,7 +1031,9 @@ def test_decode_image(self, mock_bits, mock_imread, mock_float_tensor,

steganogan.decoder.return_value.view.return_value.__gt__.return_value = _image

bits_image = _image.data.cpu.return_value.numpy.return_value.tolist.return_value
bits_image = (
_image.data.int.return_value.cpu.return_value.numpy.return_value.tolist.return_value
)
mock_bits.return_value.split.return_value = [candidate]

mock_bytearray_to_text.return_value = _candidate
Expand Down

0 comments on commit 3baf252

Please sign in to comment.