diff --git a/rocky/rocky/views/mixins.py b/rocky/rocky/views/mixins.py index e1af0bf1220..34fec039e84 100644 --- a/rocky/rocky/views/mixins.py +++ b/rocky/rocky/views/mixins.py @@ -358,10 +358,7 @@ def hydrate_report_list(self, reports: list[Report]) -> list[HydratedReport]: @staticmethod def get_total_objects(reports: list[Report]) -> int: - input_objects: set = set() - for report in reports: - input_objects.add(report.input_ooi) - return len(input_objects) + return len({report.input_ooi for report in reports}) @staticmethod def report_type_summary(reports: list[Report]) -> dict[str, int]: