Skip to content

Commit

Permalink
Improve function name
Browse files Browse the repository at this point in the history
  • Loading branch information
zhiltsov-max committed Mar 4, 2025
1 parent fd49c4c commit 1f6c046
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions src/datumaro/plugins/data_formats/coco/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
import errno
import logging as log
import os.path as osp
from collections.abc import Generator
from inspect import isclass
from typing import Any, Dict, Iterator, Optional, Tuple, Type, TypeVar, Union, overload

Expand Down Expand Up @@ -321,14 +322,14 @@ def _parse_anns(self, img_info, ann_info, item):
def _load_items(self, json_data):
pbar = self._ctx.progress_reporter

def _gen_ann(info_lists):
def _iterate_infos(info_lists: list[T]) -> Generator[T, None, None]:
while info_lists:
yield info_lists.pop()

items = {}
img_infos = {}
img_lists = self._parse_field(json_data, "images", list)
for img_info in _gen_ann(img_lists):
for img_info in _iterate_infos(img_lists):
parsed = self._parse_item(img_info)
if parsed is None:
continue
Expand All @@ -341,7 +342,7 @@ def _gen_ann(info_lists):

ann_lists = self._parse_field(json_data, "annotations", list)
for ann_info in pbar.iter(
_gen_ann(ann_lists),
_iterate_infos(ann_lists),
desc=f"Importing '{self._subset}'",
total=len(ann_lists),
):
Expand Down

0 comments on commit 1f6c046

Please sign in to comment.