fix: set TensorFlow version to 2.10.0

Kohulan · Feb 17, 2023 · 1221cad · 1221cad
1 parent 99eb533
commit 1221cad
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 3 deletions.
diff --git a/DECIMER/config.py b/DECIMER/config.py
@@ -3,7 +3,7 @@
 import efficientnet.tfkeras as efn
 import DECIMER.Efficient_Net_encoder
 import DECIMER.Transformer_decoder
-from PIL import Image
+from PIL import Image, ImageEnhance
 import numpy as np
 import io
 import cv2

diff --git a/Tests/caffeine.png b/Tests/caffeine.png
diff --git a/Tests/test_functions.py b/Tests/test_functions.py
@@ -0,0 +1,8 @@
+import pytest
+from DECIMER import predict_SMILES
+
+def get_imagetosmiles():
+	img_path = 'Tests/caffeine.png'
+	expected_result = "CN1C=NC2=C1C(=O)N(C)C(=O)N2C"
+	actual_result = predict_SMILES(image_path)
+	assert expected_result == actual_result
diff --git a/setup.py b/setup.py
@@ -19,7 +19,7 @@
     packages=setuptools.find_packages(),
     license="MIT",
     install_requires=[
-        "tensorflow>=2.8.0",
+        "tensorflow==2.10.0",
         "opencv-python",
         "pystow",
         "pyheif",

diff --git a/tox.ini b/tox.ini
@@ -1,5 +1,5 @@
 [tox]
-envlist = py37, py38, py39, py310, lint
+envlist = py38, lint
 requires = tox-conda
 
 [testenv]
@@ -41,3 +41,4 @@ exclude =
     DECIMER/efficientnetv2/effnetv2_model.py,
     DECIMER/efficientnetv2/hparams.py,
     DECIMER/efficientnetv2/utils.py,
+    Tests/test_functions.py,