diff --git a/setup.py b/setup.py
index 27cb5ad..2e532c7 100644
--- a/setup.py
+++ b/setup.py
@@ -1,6 +1,6 @@
 from setuptools import setup, find_packages
 
-REQUIRED_PKGS = ['tensorflow==2.2.1', 'scikit-learn==0.22.1']
+REQUIRED_PKGS = ['tensorflow==2.4.0', 'scikit-learn==0.22.1']
 TESTS_REQUIRE = ['pytest==5.2.2']
 
 setup(