Make torch optional (#85)

* Make torch optional

* fix tests

* Review fixes

.travis.yml

@@ -6,7 +6,7 @@ python:
 before_install:
     - sudo rm -f /etc/boto.cfg
 install:
-    - pip install .
+    - pip install .[torch]
     - pip install green
 script:
     - pip install tensorboard>=1.14 future>=0.17.1 pillow>=6.0.0 tqdm>=4.31.1 pandas>=0.24.1

README.md

@@ -21,7 +21,10 @@ pip install -r requirements.txt
 To use Transfer NLP as a library:
 
 ```
+# to install the experiment builder only
 pip install transfernlp
+# to install Transfer NLP with PyTorch and Transfer Learning in NLP support
+pip install transfernlp[torch]
 ```
 or 
 ```

setup.py

@@ -5,24 +5,27 @@
     name='transfer_nlp',
     packages=find_packages(exclude=["*.tests", "*.tests.*",
                                     "tests.*", "tests"]),
-    version='0.1.5',
+    version='0.1.6',
     license='MIT',
     description='NLP library designed for flexible research and development',
     long_description=open("README.md", "r", encoding='utf-8').read(),
     long_description_content_type='text/markdown',
     author='Peter Martigny',
     author_email='peter.martigny@gmail.com',
     url='https://github.com/feedly/transfer-nlp',
-    download_url='https://github.com/feedly/transfer-nlp/archive/v0.1.5.tar.gz',
+    download_url='https://github.com/feedly/transfer-nlp/archive/v0.1.6.tar.gz',
     keywords=['NLP', 'transfer learning', 'language models', 'NLU'],
     install_requires=[
         'numpy>=1.16.2',
-        'torch==1.1.0',
-        'pytorch-ignite>=0.2.0',
-        'smart_open>=1.8.1',
         'pyaml>=19.4.1',
         'toml>=0.10.0'
     ],
+    extras_require={
+        'torch': [
+            'torch>=1.1.0',
+            'pytorch-ignite>=0.2.0',
+        ]
+    },
     classifiers=[
         'Development Status :: 3 - Alpha',
         'Intended Audience :: Science/Research',

tests/runner/test_experiment_runner.py

@@ -11,7 +11,7 @@
 
 from transfer_nlp.plugins.config import register_plugin, ExperimentConfig
 from transfer_nlp.plugins.reporters import ReporterABC
-from transfer_nlp.plugins.trainers import TrainerABC
+from transfer_nlp.plugins.trainer_abc import TrainerABC
 from transfer_nlp.runner.experiment_runner import ExperimentRunner
 
 logger = logging.getLogger(__name__)