Skip to content

Commit

Permalink
Updated installation
Browse files Browse the repository at this point in the history
  • Loading branch information
max-c-lim committed Jan 20, 2025
1 parent 4f0bee9 commit a38d2f8
Showing 1 changed file with 5 additions and 12 deletions.
17 changes: 5 additions & 12 deletions src/spikeinterface/sorters/external/rt_sort.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,16 +116,12 @@ class RTSortSorter(BaseSorter):
For more information see https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0312438"""

installation_mesg = f"""\nTo use RTSort run:\n
>>> pip install git+https://github.com/braingeneers/braindance
>>> pip install git+https://github.com/braingeneers/braindance#egg=braindance[rt-sort]
And install the following additional dependencies:
>>> pip install {'torch' if os.name == 'nt' else 'torch_tensorrt'}
>>> pip install diptest
>>> pip install pynvml
>>> pip install scikit-learn
Additionally, install PyTorch (https://pytorch.org/get-started/locally/) with any version of CUDA as the compute platform.
If running on a Linux machine, install Torch-TensorRT (https://pytorch.org/TensorRT/getting_started/installation.html) for faster computations.
More information on RTSort at:
*https://github.com/braingeneers/braindance
More information on RTSort at: https://github.com/braingeneers/braindance
"""

handle_multi_segment = False
Expand All @@ -134,10 +130,7 @@ class RTSortSorter(BaseSorter):
def get_sorter_version(cls):
import braindance

try:
return braindance.__version__
except AttributeError:
return "0.0.1"
return braindance.__version__

@classmethod
def is_installed(cls):
Expand Down

0 comments on commit a38d2f8

Please sign in to comment.