Bump up tensorboard dependency version # (#1663)

* Update tensorboard version
* Update pyspark version
* Update mypy version

Author: hardianlawi

requirements-pyspark.txt

@@ -1,3 +1,3 @@
 # Note: we don't include PySpark in the normal required installs.
 # Installing a new version may overwrite your existing system install.
-pyspark==2.4.3
+pyspark==3.1.2

requirements.txt

@@ -25,7 +25,7 @@ munkres>=1.0.6
 networkx>=2.2,<2.6
 
 # Model introspection tools
-tensorboard>=1.14.0,<2.0.0
+tensorboard>=2.0.0,<2.7.0
 
 #### EXTRA/TEST LIBRARIES
 
@@ -46,7 +46,7 @@ dill>=0.3.0,<0.4.0
 black>=19.3b0,<20.0
 flake8>=3.7.0,<4.0.0
 isort>=4.3.0,<5.0.0
-mypy>=0.720,<0.730
+mypy>=0.800,<0.920
 pydocstyle>=4.0.0,<5.0.0
 pytest>=5.0.0,<6.0.0
 pytest-cov>=2.7.0,<3.0.0

setup.py

@@ -42,7 +42,7 @@
         "tqdm>=4.33.0,<5.0.0",
         "scikit-learn>=0.20.2,<0.25.0",
         "torch>=1.2.0,<2.0.0",
-        "tensorboard>=1.14.0,<2.0.0",
+        "tensorboard>=2.0.0,<2.7.0",
         "networkx>=2.2,<2.6",
     ],
     python_requires=">=3.6",