diff options
-rw-r--r-- | Readme.rst | 2 | ||||
-rw-r--r-- | pyspike/__init__.py | 19 |
2 files changed, 21 insertions, 0 deletions
@@ -1,6 +1,8 @@ PySpike ======= +.. image:: https://badge.fury.io/py/pyspike.png + :target: http://badge.fury.io/py/pyspike .. image:: https://travis-ci.org/mariomulansky/PySpike.svg?branch=master :target: https://travis-ci.org/mariomulansky/PySpike diff --git a/pyspike/__init__.py b/pyspike/__init__.py index a5f9f0a..3e836bd 100644 --- a/pyspike/__init__.py +++ b/pyspike/__init__.py @@ -23,3 +23,22 @@ from psth import psth from spikes import load_spike_trains_from_txt, spike_train_from_string, \ merge_spike_trains, generate_poisson_spikes + + +# define the __version__ following +# http://stackoverflow.com/questions/17583443 +from pkg_resources import get_distribution, DistributionNotFound +import os.path + +try: + _dist = get_distribution('pyspike') + # Normalize case for Windows systems + dist_loc = os.path.normcase(_dist.location) + here = os.path.normcase(__file__) + if not here.startswith(os.path.join(dist_loc, 'pyspike')): + # not installed, but there is another version that *is* + raise DistributionNotFound +except DistributionNotFound: + __version__ = 'Please install this project with setup.py' +else: + __version__ = _dist.version |