Add setup.py (#390)

Author: guillaumekln

.github/workflows/test.yml

@@ -21,8 +21,7 @@ jobs:
 
       - name: Install dependencies
         run: |
-          python -m pip install -r requirements.txt
-          python -m pip install -r test/requirements.txt
+          python -m pip install -e .[tests]
 
       - name: Check code format with Black
         run: |
@@ -35,4 +34,4 @@ jobs:
 
       - name: Test with pytest
         run: |
-          PYTHONPATH=$PWD python -m pytest test/
+          python -m pytest test/

README.md

@@ -319,10 +319,10 @@ Advanced users could extend existing frameworks to implement customized behavior
 
 ```bash
 # Install test requirements:
-pip install -r test/requirements.txt
+pip install -e .[tests]
 
 # Run unit tests:
-PYTHONPATH=$PWD pytest test/
+pytest test/
 
 # Automatically reformat code:
 black .

requirements.txt

@@ -1,4 +1,4 @@
-https://github.com/SYSTRAN/storages/archive/3601019941cca224e42dc1c1435355ebcb6d2cfd.tar.gz
+systran-storages @ https://github.com/SYSTRAN/storages/archive/3601019941cca224e42dc1c1435355ebcb6d2cfd.tar.gz
 jsonschema==3.*
 pyonmttok==1.31.0
 requests==2.*

setup.py

@@ -0,0 +1,27 @@
+import os
+
+from setuptools import find_packages, setup
+
+
+def _load_requirements(path):
+    with open(path, encoding="utf-8") as requirements:
+        return [requirement.strip() for requirement in requirements]
+
+
+base_dir = os.path.dirname(os.path.abspath(__file__))
+install_requires = _load_requirements(os.path.join(base_dir, "requirements.txt"))
+tests_require = _load_requirements(os.path.join(base_dir, "test", "requirements.txt"))
+
+setup(
+    name="nmt-wizard-docker",
+    version="0.1.0",
+    license="MIT",
+    description="Dockerization of NMT frameworks",
+    author="OpenNMT",
+    python_requires=">=3.6",
+    install_requires=install_requires,
+    extras_require={
+        "tests": tests_require,
+    },
+    packages=find_packages(include=["nmtwizard"]),
+)