Skip to content

Commit

Permalink
fix: Update TensorFlow, fix: Tests
Browse files Browse the repository at this point in the history
  • Loading branch information
Kohulan committed Feb 17, 2023
1 parent 54dbbc8 commit 1d2b583
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 5 deletions.
5 changes: 3 additions & 2 deletions Tests/test_functions.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,9 @@
import pytest
from DECIMER import predict_SMILES

def get_imagetosmiles():
@pytest.mark.filterwarnings('ignore::DeprecationWarning')
def test_imagetosmiles():
img_path = 'Tests/caffeine.png'
expected_result = "CN1C=NC2=C1C(=O)N(C(=O)N2C)C"
actual_result = predict_SMILES(image_path)
actual_result = predict_SMILES(img_path)
assert expected_result == actual_result
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
packages=setuptools.find_packages(),
license="MIT",
install_requires=[
"tensorflow==2.10.0",
"tensorflow==2.10.1",
"opencv-python",
"pystow",
"pyheif",
Expand Down
5 changes: 3 additions & 2 deletions tox.ini
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
[tox]
envlist = py38, lint
envlist = py310, lint
requires = tox-conda

[testenv]
Expand All @@ -10,7 +10,7 @@ conda_deps =
pytest
conda_channels =
conda-forge
commands = pytest --basetemp="{envtmpdir}" {posargs}
commands = pytest -p no:warnings --basetemp="{envtmpdir}" {posargs}

[testenv:lint]
basepython = python3
Expand Down Expand Up @@ -42,3 +42,4 @@ exclude =
DECIMER/efficientnetv2/hparams.py,
DECIMER/efficientnetv2/utils.py,
Tests/test_functions.py,

0 comments on commit 1d2b583

Please sign in to comment.