diff --git a/setup.py b/setup.py index 79b7a8baf7..02c69d5a7a 100755 --- a/setup.py +++ b/setup.py @@ -7,7 +7,7 @@ from setuptools import setup, find_packages # http://blog.ionelmc.ro/2014/05/25/python-packaging/ setup( name="pytorch-lightning", - version='0.2.2', + version='0.2.3', description="The Keras for ML researchers using PyTorch", author="William Falcon", author_email="waf2107@columbia.edu", @@ -19,7 +19,7 @@ setup( install_requires=[ "torch>=1.1.0", "tqdm", - "test-tube>=0.653", + "test-tube>=0.6.6", "tensorflow>=1.14.0" ], packages=find_packages(),