Merge pull request #2329 from NervanaSystems/mkarzyns/version_fix

[Py] Fix version label

Merge pull request #2329 from NervanaSystems/mkarzyns/version_fix
[Py] Fix version label
4dd9e110 · Adam Procter · GitHub · a534274f · 544aa6a3 · 4dd9e110
Unverified Commit 4dd9e110 authored Jan 16, 2019 by Adam Procter Committed by GitHub Jan 16, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 2 deletions

__init__.py python/ngraph/__init__.py +6 -2

No files found.
--- a/python/ngraph/__init__.py
+++ b/python/ngraph/__init__.py
@@ -15,8 +15,12 @@
 # ******************************************************************************
 """ngraph module namespace, exposing factory functions for all ops and other classes."""

-from pkg_resources import get_distribution
-__version__ = get_distribution('ngraph-core').version
+from pkg_resources import get_distribution, DistributionNotFound
+
+try:
+    __version__ = get_distribution('ngraph-core').version
+except DistributionNotFound:
+    __version__ = '0.0.0-dev'

 from ngraph.ops import absolute
 from ngraph.ops import absolute as abs