fixed bug in transform_featurepoints where feature IDs are converted …

…to floats and causes statistic calculation to fail
tobac-project · Nov 7, 2023 · 09adaf2 · 09adaf2
1 parent 80ef32c
commit 09adaf2
Show file tree

Hide file tree

Showing 3 changed files with 63 additions and 50 deletions.
diff --git a/...les/Example_Track_on_Radar_Segment_on_Satellite/track_on_radar_segment_on_satellite.ipynb b/...les/Example_Track_on_Radar_Segment_on_Satellite/track_on_radar_segment_on_satellite.ipynb
diff --git a/tobac/utils/bulk_statistics.py b/tobac/utils/bulk_statistics.py
@@ -43,8 +43,8 @@ def get_statistics(
         Dictionary with function(s) to apply over each region as values and the name of the respective statistics as keys
         default is to just count the number of cells associated with each feature and write it to the feature dataframe
     index: None | list[int], optional (default: None)
-        list of indexes of regions in labels to apply function to. If None, will
-            default to all integers between the minimum and the maximum value in labels
+        list of indices of regions in labels to apply function to. If None, will
+            default to all integer feature labels in labels
     default: None | float, optional (default: None)
         default value to return in a region that has no values
     id_column: str, optional (default: "feature")

diff --git a/tobac/utils/general.py b/tobac/utils/general.py
@@ -787,6 +787,9 @@ def transform_feature_points(
             "Dropping feature numbers: " + str(removed_features.values), UserWarning
         )
 
+    # make sure that feature points are converted back to int64
+    ret_features['feature'] = ret_features.feature.astype(int)
+
     return ret_features