diff --git a/background_subtraction_heat.py b/background_subtraction_heat.py index d712c07..62495b6 100644 --- a/background_subtraction_heat.py +++ b/background_subtraction_heat.py @@ -53,9 +53,9 @@ class RecordingGrouper: self.data = data def getRecordings(self): - for uc in self.data.use_cases(): - for detection in data.detections(uc): - return detection.recordings() + for detection in data.detections(self.data.use_cases()): + for recording in detection.recordings(): + yield recording def getRecordingGroups(self): groups = []