Skip to content

Commit

Permalink
[FIx] Fix ade20k dataset wrong return typing (#10449)
Browse files Browse the repository at this point in the history
  • Loading branch information
xiexinch committed Jun 6, 2023
1 parent 28c698c commit f2f8925
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions mmdet/datasets/ade20k.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
# Copyright (c) OpenMMLab. All rights reserved.
import os.path as osp
from typing import List

from mmengine import fileio

Expand Down Expand Up @@ -220,11 +221,11 @@ def __init__(self,
super().__init__(
img_suffix=img_suffix, seg_map_suffix=seg_map_suffix, **kwargs)

def load_data_list(self) -> list[dict]:
def load_data_list(self) -> List[dict]:
"""Load annotation from directory or annotation file.
Returns:
list[dict]: All data info of dataset.
List[dict]: All data info of dataset.
"""
data_list = []
img_dir = self.data_prefix.get('img_path', None)
Expand Down

0 comments on commit f2f8925

Please sign in to comment.