From 1285168f67cf1fb80496a325f089bc5aecffa7ae Mon Sep 17 00:00:00 2001 From: Vinnam Kim Date: Wed, 25 Oct 2023 17:02:16 +0900 Subject: [PATCH] Hotfix DatasetEntity.get_combined_subset function loop (#2577) Fix get_combined_subset function --- src/otx/api/entities/datasets.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/otx/api/entities/datasets.py b/src/otx/api/entities/datasets.py index 1a37458ebe9..ff8660154ca 100644 --- a/src/otx/api/entities/datasets.py +++ b/src/otx/api/entities/datasets.py @@ -349,8 +349,9 @@ def get_combined_subset(self, subsets: List[Subset]) -> "DatasetEntity": Returns: DatasetEntity: DatasetEntity with items matching subsets """ + to_keep = set(subsets) dataset = DatasetEntity( - items=[item for item in self._items if item.subset in set(subsets)], + items=[item for item in self if item.subset in to_keep], purpose=self.purpose, ) return dataset