Merge pull request #62 from microsoft/jsleep/torch_version

adding torch as a required depedency with capped version

Author: shatu

VERSION_NOTES.md

@@ -1,4 +1,7 @@
 # Version Notes:
+## 0.2.7
+* Adding torch<=1.9 as required dependency
+
 ## 0.2.6
 * Adding support for parsing multi-level args from commandline and params
 

setup.py

@@ -3,16 +3,15 @@
 with open("README.md", "r") as fh:
     long_description = fh.read()
 
-# please install torch separately
-required = ['tqdm','tensorboard', 'Pillow','azureml-core','pyyaml','pandas']
+required = ['torch<=1.9','tqdm','tensorboard', 'Pillow','azureml-core','pyyaml','pandas']
 extras = {
     'dev': ['pylint', 'pytest', 'pytest-cov'],
     'plugins': ['transformers','pandas','matplotlib','sklearn','scipy','rouge-score']
 }
 
 setup(
     name="pymarlin",
-    version="0.2.6",
+    version="0.2.7",
     author="ELR Team",
     author_email="[email protected]",
     description="Lightweight Deeplearning Library",