From 5f5bbbdd86a5e2962a37ef8ab477c89ebe9f0a26 Mon Sep 17 00:00:00 2001
From: Fedor Ignatov <ignatov.fedor@gmail.com>
Date: Mon, 27 Apr 2020 12:42:51 +0300
Subject: [PATCH] fix: requirements in nemo configs + add torchvision to
 pytorch.txt (#1186)

* fix: requirements in nemo configs + added torchvision to pytorch.txt

* refactor: latest pytorch and torchvision
---
 deeppavlov/__init__.py                    | 2 +-
 deeppavlov/configs/nemo/asr.json          | 1 +
 deeppavlov/configs/nemo/asr_tts.json      | 1 +
 deeppavlov/configs/nemo/tts.json          | 1 +
 deeppavlov/requirements/pytorch.txt       | 3 ++-
 tests/test_configs/nemo/tts2asr_test.json | 1 +
 6 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/deeppavlov/__init__.py b/deeppavlov/__init__.py
index 40938d648f..41d8ba0982 100644
--- a/deeppavlov/__init__.py
+++ b/deeppavlov/__init__.py
@@ -39,7 +39,7 @@ def evaluate_model(config: [str, Path, dict], download: bool = False, recursive:
 except ImportError:
     'Assuming that requirements are not yet installed'
 
-__version__ = '0.9.0'
+__version__ = '0.9.1'
 __author__ = 'Neural Networks and Deep Learning lab, MIPT'
 __description__ = 'An open source library for building end-to-end dialog systems and training chatbots.'
 __keywords__ = ['NLP', 'NER', 'SQUAD', 'Intents', 'Chatbot']
diff --git a/deeppavlov/configs/nemo/asr.json b/deeppavlov/configs/nemo/asr.json
index 402d258337..f54af6c43a 100644
--- a/deeppavlov/configs/nemo/asr.json
+++ b/deeppavlov/configs/nemo/asr.json
@@ -17,6 +17,7 @@
       "NEMO_PATH": "~/.deeppavlov/models/nemo"
     },
     "requirements": [
+      "{DEEPPAVLOV_PATH}/requirements/pytorch.txt",
       "{DEEPPAVLOV_PATH}/requirements/nemo-asr.txt"
     ],
     "download": [
diff --git a/deeppavlov/configs/nemo/asr_tts.json b/deeppavlov/configs/nemo/asr_tts.json
index 5335bb463f..6ae19248d5 100644
--- a/deeppavlov/configs/nemo/asr_tts.json
+++ b/deeppavlov/configs/nemo/asr_tts.json
@@ -35,6 +35,7 @@
       "TTS_PATH": "{NEMO_PATH}/tacotron2_waveglow"
     },
     "requirements": [
+      "{DEEPPAVLOV_PATH}/requirements/pytorch.txt",
       "{DEEPPAVLOV_PATH}/requirements/nemo-asr.txt",
       "{DEEPPAVLOV_PATH}/requirements/nemo-tts.txt"
     ],
diff --git a/deeppavlov/configs/nemo/tts.json b/deeppavlov/configs/nemo/tts.json
index 8973604539..dfeb13067d 100644
--- a/deeppavlov/configs/nemo/tts.json
+++ b/deeppavlov/configs/nemo/tts.json
@@ -18,6 +18,7 @@
       "TTS_PATH": "{NEMO_PATH}/tacotron2_waveglow"
     },
     "requirements": [
+      "{DEEPPAVLOV_PATH}/requirements/pytorch.txt",
       "{DEEPPAVLOV_PATH}/requirements/nemo-asr.txt",
       "{DEEPPAVLOV_PATH}/requirements/nemo-tts.txt"
     ],
diff --git a/deeppavlov/requirements/pytorch.txt b/deeppavlov/requirements/pytorch.txt
index 7657e148b1..6aea1bd158 100644
--- a/deeppavlov/requirements/pytorch.txt
+++ b/deeppavlov/requirements/pytorch.txt
@@ -1 +1,2 @@
-torch==1.4.0
\ No newline at end of file
+torch==1.5.0
+torchvision==0.6.0
\ No newline at end of file
diff --git a/tests/test_configs/nemo/tts2asr_test.json b/tests/test_configs/nemo/tts2asr_test.json
index 22bc170143..c97cb1eaf7 100644
--- a/tests/test_configs/nemo/tts2asr_test.json
+++ b/tests/test_configs/nemo/tts2asr_test.json
@@ -36,6 +36,7 @@
       "TTS_PATH": "{NEMO_PATH}/tacotron2_waveglow"
     },
     "requirements": [
+      "{DEEPPAVLOV_PATH}/requirements/pytorch.txt",
       "{DEEPPAVLOV_PATH}/requirements/nemo-asr.txt",
       "{DEEPPAVLOV_PATH}/requirements/nemo-tts.txt"
     ],