Merge pull request #693 from EmmaRenauld/fix_warning

Fixing warning always showing up.
scilus · Mar 16, 2023 · bb5c4cb · bb5c4cb
2 parents cdbccac + d020289
commit bb5c4cb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/scilpy/tracking/tools.py b/scilpy/tracking/tools.py
@@ -223,9 +223,9 @@ def _warn_and_save(new_streamlines, sft):
     """Last step of the two resample functions:
     Warn that we loose data_per_point, then create resampled SFT."""
 
-    if sft.data_per_point is not None:
-        logging.debug("Initial stateful tractogram contained data_per_point. "
-                      "This information will not be carried in the final"
+    if sft.data_per_point is not None and sft.data_per_point.keys():
+        logging.debug("Initial StatefulTractogram contained data_per_point. "
+                      "This information will not be carried in the final "
                       "tractogram.")
     new_sft = StatefulTractogram.from_sft(
         new_streamlines, sft, data_per_streamline=sft.data_per_streamline)

diff --git a/scilpy/utils/streamlines.py b/scilpy/utils/streamlines.py
@@ -242,9 +242,9 @@ def compress_sft(sft, tol_error=0.01):
     # Compress streamlines
     compressed_streamlines = compress_streamlines(sft.streamlines,
                                                   tol_error=tol_error)
-    if sft.data_per_point is not None:
+    if sft.data_per_point is not None and sft.data_per_point.keys():
         logging.warning("Initial StatefulTractogram contained data_per_point. "
-                        "This information will not be carried in the final"
+                        "This information will not be carried in the final "
                         "tractogram.")
 
     compressed_sft = StatefulTractogram.from_sft(