setup: Update TF version to fix vulnerabilities

cabrust · Sep 1, 2021 · 71a4e03 · 71a4e03
1 parent 87a82ca
commit 71a4e03
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -33,8 +33,8 @@ CHIA depends on:
 * gputil ~= 1.4.0
 * networkx ~= 2.4
 * numpy ~= 1.18.5
-* tensorflow-addons == 0.11.1
-* tensorflow == 2.4.0
+* tensorflow-addons == 0.14.0
+* tensorflow == 2.4.3
 
 ## Installation
 To install, simply run:

diff --git a/setup.py b/setup.py
@@ -26,8 +26,8 @@
         "gputil~=1.4.0",
         "networkx~=2.4",
         "numpy~=1.19.2",
-        "tensorflow-addons==0.12.1",
-        "tensorflow~=2.4.0",
+        "tensorflow-addons==0.14.0",
+        "tensorflow~=2.4.3",
     ],
     # metadata to display on PyPI
     author="Clemens-Alexander Brust",
@@ -39,6 +39,7 @@
         "License :: OSI Approved :: BSD License",
         "Programming Language :: Python :: 3.7",
         "Programming Language :: Python :: 3.8",
+        "Programming Language :: Python :: 3.9",
         "Development Status :: 5 - Production/Stable",
         "Environment :: GPU",
         "Operating System :: POSIX :: Linux",