From 5f1bc73d4c109cfaac31da76321e6d626aa301dc Mon Sep 17 00:00:00 2001 From: Benjamin Midtvedt Date: Wed, 15 Feb 2023 15:35:26 +0100 Subject: [PATCH] Fix issue with repeated oneof-features --- deeptrack/features.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/deeptrack/features.py b/deeptrack/features.py index ad4998cbb..14beec481 100644 --- a/deeptrack/features.py +++ b/deeptrack/features.py @@ -1307,8 +1307,8 @@ def _process_properties(self, propertydict) -> dict: return propertydict - def get(self, image, key, **kwargs): - return self.collection[key](image) + def get(self, image, key, _ID=(), **kwargs): + return self.collection[key](image, _ID=_ID) class OneOfDict(Feature): @@ -1338,8 +1338,8 @@ def _process_properties(self, propertydict) -> dict: return propertydict - def get(self, image, key, **kwargs): - return self.collection[key](image) + def get(self, image, key, _ID=(), **kwargs): + return self.collection[key](image, _ID=_ID) # class Dataset(Feature): @@ -2109,7 +2109,7 @@ def _check_volumes_non_overlapping(self, volume_1, volume_2, min_distance): ) def _resample_volume_position(self, volume): - """ Draws a new position for the volume. """ + """Draws a new position for the volume.""" for pdict in volume.properties: if "position" in pdict and "_position_sampler" in pdict: @@ -2122,4 +2122,4 @@ def _resample_volume_position(self, volume): # Alias -Dataset = TensorflowDataset \ No newline at end of file +Dataset = TensorflowDataset