Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
plyfager committed Dec 13, 2022
1 parent 86492f8 commit b826a3f
Showing 1 changed file with 7 additions and 9 deletions.
16 changes: 7 additions & 9 deletions tests/test_datasets/test_mscoco_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,18 +16,16 @@ def test_mscoco(self):
# test basic usage
dataset = MSCoCoDataset(data_root=self.data_root, pipeline=[])
assert dataset[0] == dict(
gt_label="a good meal",
img_path=os.path.join(
self.data_root,
'train2014/COCO_train2014_000000000009.jpg'),
gt_label='a good meal',
img_path=os.path.join(self.data_root,
'train2014/COCO_train2014_000000000009.jpg'),
sample_idx=0)

# test with different phase
dataset = MSCoCoDataset(
data_root=self.data_root, phase="val", pipeline=[])
data_root=self.data_root, phase='val', pipeline=[])
assert dataset[0] == dict(
gt_label="a pair of slippers",
img_path=os.path.join(
self.data_root,
'val2014/COCO_val2014_000000000042.jpg'),
gt_label='a pair of slippers',
img_path=os.path.join(self.data_root,
'val2014/COCO_val2014_000000000042.jpg'),
sample_idx=0)

0 comments on commit b826a3f

Please sign in to comment.