Skip to content

Commit 2ee6f33

Browse files
committed
feat: fix format
Signed-off-by: Kaan Çolak <kaancolak95@gmail.com>
1 parent 9b7275f commit 2ee6f33

File tree

2 files changed

+9
-9
lines changed

2 files changed

+9
-9
lines changed

projects/AutowareCenterPoint/tests/test_pillar_encorder_autoware.py

+7-4
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,11 @@
11
import pytest
22
import torch
3-
import logging
43

54
from mmdet3d.registry import MODELS
5+
from projects.AutowareCenterPoint.centerpoint.pillar_encoder_autoware import \
6+
PillarFeatureNetAutoware # noqa: F401
7+
68

7-
import projects.AutowareCenterPoint.centerpoint.pillar_encoder_autoware
89
def test_pillar_feature_net_autoware():
910

1011
use_voxel_center_z = False
@@ -29,9 +30,11 @@ def test_pillar_feature_net_autoware():
2930
features = pillar_feature_net_autoware(features, num_voxels, coors)
3031

3132
if not use_voxel_center_z:
32-
assert pillar_feature_net_autoware.pfn_layers[0].linear.in_features == 9
33+
assert pillar_feature_net_autoware.pfn_layers[
34+
0].linear.in_features == 9
3335
else:
34-
assert pillar_feature_net_autoware.pfn_layers[0].linear.in_features == 9
36+
assert pillar_feature_net_autoware.pfn_layers[
37+
0].linear.in_features == 9
3538

3639
assert pillar_feature_net_autoware.pfn_layers[0].linear.out_features == 64
3740

projects/AutowareCenterPoint/tests/test_t4_dataset.py

+2-5
Original file line numberDiff line numberDiff line change
@@ -3,17 +3,14 @@
33
from mmengine.registry import TRANSFORMS
44
from mmengine.structures import InstanceData
55

6-
from mmdet3d.datasets import NuScenesDataset
76
from mmdet3d.structures import Det3DDataSample, LiDARInstance3DBoxes
8-
97
from projects.AutowareCenterPoint.datasets.tier4_dataset import T4Dataset
108

9+
1110
def _generate_t4_dataset_config():
1211
data_root = 'data/sample_dataset/'
1312
ann_file = 'T4Dataset_infos_train.pkl'
14-
classes = [
15-
'car', 'truck', 'bus', 'bicycle', 'pedestrian'
16-
]
13+
classes = ['car', 'truck', 'bus', 'bicycle', 'pedestrian']
1714

1815
if 'Identity' not in TRANSFORMS:
1916

0 commit comments

Comments
 (0)