Add dependency of pytorch and ignite back (#145)

* fix lfs * dependency on pytorch-ignite-nightly

Add dependency of pytorch and ignite back (#145)
* fix lfs * dependency on pytorch-ignite-nightly
8bc1f96a · Gao, Xiang · GitHub · c5cfa898 · 8bc1f96a · 8bc1f96a
Unverified Commit 8bc1f96a authored Dec 09, 2018 by Gao, Xiang Committed by GitHub Dec 09, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 109 deletions

azure/install_dependencies.sh azure/install_dependencies.sh +0 -1

setup.py setup.py +2 -2

tests/test_data/inputtrain.yaml tests/test_data/inputtrain.yaml +3 -106

No files found.
--- a/azure/install_dependencies.sh
+++ b/azure/install_dependencies.sh
@@ -3,4 +3,3 @@
 python -m pip install --upgrade pip
 pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
 pip install tqdm ase tensorboardX pyyaml
-pip install pytorch-ignite --no-deps
--- a/setup.py
+++ b/setup.py
@@ -11,8 +11,8 @@ setup_attrs = {
    'packages': find_packages(),
    'include_package_data': True,
    'install_requires': [
-        # 'torch-nightly',
+        'torch-nightly',
-        # 'pytorch-ignite',
+        'pytorch-ignite-nightly',
        'lark-parser',
        'h5py',
    ],

--- a/tests/test_data/inputtrain.yaml
+++ b/tests/test_data/inputtrain.yaml