[Fix] Fix DATASETS not imported error (#1827)

pull/1829/head
MengzhangLI 2022-07-26 14:46:59 +08:00 committed by GitHub
parent 3cf2bbda2a
commit 6758b148b3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions

View File

@ -6,10 +6,10 @@ from unittest.mock import MagicMock
import pytest
from mmseg.datasets import (ADE20KDataset, BaseSegDataset,
CityscapesDataset, COCOStuffDataset, ISPRSDataset,
LoveDADataset, PascalVOCDataset, PotsdamDataset,
iSAIDDataset)
from mmseg.datasets import (ADE20KDataset, BaseSegDataset, CityscapesDataset,
COCOStuffDataset, ISPRSDataset, LoveDADataset,
PascalVOCDataset, PotsdamDataset, iSAIDDataset)
from mmseg.registry import DATASETS
from mmseg.utils import get_classes, get_palette