Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

Commit

Permalink
fix data outpye type
Browse files Browse the repository at this point in the history
  • Loading branch information
edgarriba committed Apr 20, 2021
1 parent 32ac9b3 commit 176634c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion flash/vision/classification/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,7 @@ def _load_data_dir(cls, data: Any, dataset: Optional[AutoDataset] = None) -> Lis
# labels might not contain the complete set of ids so that you can infer the total
# number of classes to train in your dataset.
dataset.num_classes = len(data)
out: List[str, int] = []
out: List[Tuple[str, int]] = []
for p, label in data:
if os.path.isdir(p):
files_list: List[str] = os.listdir(p)
Expand Down

0 comments on commit 176634c

Please sign in to comment.