Release v1.4.1 of NNCF to master

Meshford · Jul 28, 2020 · 58b7bb6 · 58b7bb6
2 parents 07f56d9 + 837ccd2
commit 58b7bb6
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 12 deletions.
diff --git a/README.md b/README.md
@@ -82,6 +82,13 @@ See [third_party_integration](./third_party_integration) for examples of code mo
 ## Installation
 We suggest to install or use the package in the [Python virtual environment](https://docs.python.org/3/tutorial/venv.html).
 
+#### As a PyPI package:
+NNCF can be installed as a regular PyPI package via pip:
+```
+sudo apt install python3-dev
+pip install nncf
+```
+
 #### As a package built from checked-out repository:
 1) Install the following system dependencies:
 

diff --git a/nncf/version.py b/nncf/version.py
@@ -1 +1 @@
-__version__ = "1.4"
+__version__ = "1.4.1"
diff --git a/setup.py b/setup.py
@@ -147,7 +147,9 @@ def write_install_type(install_type: str):
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://github.com/openvinotoolkit/nncf_pytorch",
-    packages=setuptools.find_packages(),
+    packages=setuptools.find_packages(exclude=["tests", "tests.*",
+                                               "examples", "examples.*",
+                                               "tools", "tools.*"]),
     dependency_links=DEPENDENCY_LINKS,
     classifiers=[
         "Programming Language :: Python :: 3",

diff --git a/tests/test_install.py b/tests/test_install.py
@@ -20,7 +20,7 @@
 INSTALL_CHECKS_FILENAME = 'install_checks.py'
 
 
-@pytest.fixture(name="package_type", params=["install", "develop", "sdist", "bdist_wheel"])
+@pytest.fixture(name="package_type", params=["install", "develop", "sdist", "bdist_wheel", "pypi"])
 def package_type_(request):
     return request.param
 
@@ -41,15 +41,21 @@ def test_install(install_type, tmp_path, package_type):
 
     shutil.copy(TEST_ROOT / INSTALL_CHECKS_FILENAME, run_path)
 
-    subprocess.run(
-        "{python} {nncf_repo_root}/setup.py {package_type} {install_flag}".format(python=python_executable_with_venv,
-                                                                                  nncf_repo_root=PROJECT_ROOT,
-                                                                                  package_type=package_type,
-                                                                                  install_flag='--cpu-only' if
-                                                                                  install_type == "CPU" else ''),
-        check=True,
-        shell=True,
-        cwd=PROJECT_ROOT)
+    if package_type == "pypi":
+        subprocess.run(
+            "{} install nncf".format(pip_with_venv), check=True, shell=True)
+    else:
+
+        subprocess.run(
+            "{python} {nncf_repo_root}/setup.py {package_type} {install_flag}".format(
+                python=python_executable_with_venv,
+                nncf_repo_root=PROJECT_ROOT,
+                package_type=package_type,
+                install_flag='--cpu-only' if
+                install_type == "CPU" else ''),
+            check=True,
+            shell=True,
+            cwd=PROJECT_ROOT)
 
     # Do additional install step for sdist/bdist packages
     if package_type == "sdist":