Reorganized nemo as a PyPI package (pulp-platform#3)

* Reorganized nemo as a PIP package * Added install_requires requirements to setup.py
FrancescoConti · Apr 9, 2020 · 567e514 · 567e514
1 parent 89da083
commit 567e514
Show file tree

Hide file tree

Showing 21 changed files with 30 additions and 1 deletion.
diff --git a/.github/workflows/nemo.yml b/.github/workflows/nemo.yml
@@ -38,4 +38,4 @@ jobs:
     - name: Test
       run: |
         # pip install pytest
-        cd tests; wget https://raw.githubusercontent.com/FrancescoConti/nemo_examples_helper/master/mnist_cnn_fp.pt; PYTHONPATH=`pwd`/../.. python mnist_test.py
+        cd tests; wget https://raw.githubusercontent.com/FrancescoConti/nemo_examples_helper/master/mnist_cnn_fp.pt; PYTHONPATH=`pwd`/.. python mnist_test.py
diff --git a/__init__.py → nemo/__init__.py b/__init__.py → nemo/__init__.py
diff --git a/evaluation.py → nemo/evaluation.py b/evaluation.py → nemo/evaluation.py
diff --git a/graph.py → nemo/graph.py b/graph.py → nemo/graph.py
diff --git a/precision.py → nemo/precision.py b/precision.py → nemo/precision.py
diff --git a/quant/__init__.py → nemo/quant/__init__.py b/quant/__init__.py → nemo/quant/__init__.py
diff --git a/quant/pact.py → nemo/quant/pact.py b/quant/pact.py → nemo/quant/pact.py
diff --git a/relaxation.py → nemo/relaxation.py b/relaxation.py → nemo/relaxation.py
diff --git a/transf/__init__.py → nemo/transf/__init__.py b/transf/__init__.py → nemo/transf/__init__.py
diff --git a/transf/bias.py → nemo/transf/bias.py b/transf/bias.py → nemo/transf/bias.py
diff --git a/transf/bn.py → nemo/transf/bn.py b/transf/bn.py → nemo/transf/bn.py
diff --git a/transf/common.py → nemo/transf/common.py b/transf/common.py → nemo/transf/common.py
diff --git a/transf/deploy.py → nemo/transf/deploy.py b/transf/deploy.py → nemo/transf/deploy.py
diff --git a/transf/equalize.py → nemo/transf/equalize.py b/transf/equalize.py → nemo/transf/equalize.py
diff --git a/transf/export.py → nemo/transf/export.py b/transf/export.py → nemo/transf/export.py
diff --git a/transf/pruning.py → nemo/transf/pruning.py b/transf/pruning.py → nemo/transf/pruning.py
diff --git a/transf/statistics.py → nemo/transf/statistics.py b/transf/statistics.py → nemo/transf/statistics.py
diff --git a/transf/utils.py → nemo/transf/utils.py b/transf/utils.py → nemo/transf/utils.py
diff --git a/transform.py → nemo/transform.py b/transform.py → nemo/transform.py
diff --git a/utils.py → nemo/utils.py b/utils.py → nemo/utils.py
diff --git a/setup.py b/setup.py
@@ -0,0 +1,29 @@
+import setuptools
+
+with open("README.md", "r") as fh:
+    long_description = fh.read()
+
+setuptools.setup(
+    name="NEMO",
+    version="0.1",
+    author="Francesco Conti",
+    author_email="[email protected]",
+    description="NEural Minimizer for pytOrch",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    url="https://github.com/pulp-platform/nemo",
+    packages=setuptools.find_packages(),
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: Apache Software License",
+        "Operating System :: OS Independent",
+    ],
+    python_requires='>=3.5',
+    install_requires=[
+        "torch>=1.3.0",
+        "torchvision>=0.4.1",
+        "numpy",
+        "tqdm",
+        "packaging"
+    ]
+)