diff --git a/dwi_ml/data/processing/streamlines/post_processing.py b/dwi_ml/data/processing/streamlines/post_processing.py index 67bed921..0d9f424c 100644 --- a/dwi_ml/data/processing/streamlines/post_processing.py +++ b/dwi_ml/data/processing/streamlines/post_processing.py @@ -5,9 +5,10 @@ import numpy as np import torch -from scilpy.tractograms.uncompress import uncompress + from scilpy.tractanalysis.tools import \ extract_longest_segments_from_profile as segmenting_func +from scilpy.tractanalysis.uncompress import uncompress # We could try using nan instead of zeros for non-existing previous dirs... DEFAULT_UNEXISTING_VAL = torch.zeros((1, 3), dtype=torch.float32) diff --git a/requirements.txt b/requirements.txt index 652be44a..38082530 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,11 +1,8 @@ # Supported for python 3.10 # Should work for python > 3.8. -# ---------- -# Scilpy and dipy must be installed manually first. -# (or run pip install -r requirements_github.txt) -# Not adding here to let you use your favorite version. -# ---------- +scilpy==1.5.0 + bertviz~=1.4.0 # For transformer's visu torch==1.13.* diff --git a/requirements_github.txt b/requirements_github.txt deleted file mode 100644 index 078cf9dd..00000000 --- a/requirements_github.txt +++ /dev/null @@ -1,10 +0,0 @@ - -# --------------- -# Note to developpers -# Could not manage to import hot dipy on github. -# We had to use Emmanuelle Renauld's branch of scilpy, with hot dipy skipped. - -# Not installing dipy. Will be installed trough my scilpy. -git+https://github.com/EmmaRenauld/scilpy.git@BRANCH_OLD_DIPY#egg=scilpy -# Will install dipy through requirements. -