Skip to content

Commit 0a731a0

Browse files
zhouyizhuang-megviiJianfeng Wang
authored andcommitted
fix(classification): fix pylint (#77)
* fix(classification): fix pylint * fix(classification): isort
1 parent 6494f83 commit 0a731a0

File tree

7 files changed

+29
-44
lines changed

7 files changed

+29
-44
lines changed

official/vision/classification/resnet/inference.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,15 +9,14 @@
99
import argparse
1010
import json
1111

12-
import model as resnet_model
12+
import model as resnet_model # pylint-disable=import-error
1313

1414
import cv2
1515
import numpy as np
1616

1717
import megengine
1818
import megengine.data.transform as T
1919
import megengine.functional as F
20-
import megengine.jit as jit
2120

2221
logging = megengine.logger.get_logger()
2322

official/vision/classification/resnet/test.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -7,20 +7,16 @@
77
# software distributed under the License is distributed on an
88
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
99
import argparse
10-
import bisect
1110
import multiprocessing
12-
import os
13-
import threading
1411
import time
1512

16-
import model as resnet_model
13+
import model as resnet_model # pylint-disable=import-error
1714

1815
import megengine
19-
from megengine import data as data
20-
from megengine import distributed as dist
21-
from megengine import functional as F
22-
from megengine import jit as jit
23-
from megengine.data import transform as T
16+
import megengine.data as data
17+
import megengine.data.transform as T
18+
import megengine.distributed as dist
19+
import megengine.functional as F
2420

2521
logging = megengine.logger.get_logger()
2622

official/vision/classification/resnet/train.py

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -10,19 +10,17 @@
1010
import bisect
1111
import multiprocessing
1212
import os
13-
import threading
1413
import time
1514

16-
import model as resnet_model
15+
import model as resnet_model # pylint-disable=import-error
1716

1817
import megengine
19-
from megengine import autodiff as autodiff
20-
from megengine import data as data
21-
from megengine import distributed as dist
22-
from megengine import functional as F
23-
from megengine import jit as jit
24-
from megengine import optimizer as optim
25-
from megengine.data import transform as T
18+
import megengine.autodiff as autodiff
19+
import megengine.data as data
20+
import megengine.data.transform as T
21+
import megengine.distributed as dist
22+
import megengine.functional as F
23+
import megengine.optimizer as optim
2624

2725
logging = megengine.logger.get_logger()
2826

official/vision/classification/shufflenet/inference.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,15 +9,14 @@
99
import argparse
1010
import json
1111

12-
import model as snet_model
12+
import model as snet_model # pylint-disable=import-error
1313

1414
import cv2
1515
import numpy as np
1616

1717
import megengine
1818
import megengine.data.transform as T
1919
import megengine.functional as F
20-
import megengine.jit as jit
2120

2221
logging = megengine.logger.get_logger()
2322

official/vision/classification/shufflenet/model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,9 +33,9 @@
3333
# This file has been modified by Megvii ("Megvii Modifications").
3434
# All Megvii Modifications are Copyright (C) 2014-2019 Megvii Inc. All rights reserved.
3535
# ------------------------------------------------------------------------------
36-
from megengine import functional as F
37-
from megengine import hub as hub
38-
from megengine import module as M
36+
import megengine.functional as F
37+
import megengine.hub as hub
38+
import megengine.module as M
3939

4040

4141
class ShuffleV2Block(M.Module):

official/vision/classification/shufflenet/test.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -7,20 +7,16 @@
77
# software distributed under the License is distributed on an
88
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
99
import argparse
10-
import bisect
1110
import multiprocessing
12-
import os
13-
import threading
1411
import time
1512

16-
import model as snet_model
13+
import model as snet_model # pylint-disable=import-error
1714

1815
import megengine
19-
from megengine import data as data
20-
from megengine import distributed as dist
21-
from megengine import functional as F
22-
from megengine import jit as jit
23-
from megengine.data import transform as T
16+
import megengine.data as data
17+
import megengine.data.transform as T
18+
import megengine.distributed as dist
19+
import megengine.functional as F
2420

2521
logging = megengine.logger.get_logger()
2622

official/vision/classification/shufflenet/train.py

Lines changed: 7 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -7,22 +7,19 @@
77
# software distributed under the License is distributed on an
88
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
99
import argparse
10-
import bisect
1110
import multiprocessing
1211
import os
13-
import threading
1412
import time
1513

16-
import model as snet_model
14+
import model as snet_model # pylint-disable=import-error
1715

1816
import megengine
19-
from megengine import autodiff as autodiff
20-
from megengine import data as data
21-
from megengine import distributed as dist
22-
from megengine import functional as F
23-
from megengine import jit as jit
24-
from megengine import optimizer as optim
25-
from megengine.data import transform as T
17+
import megengine.autodiff as autodiff
18+
import megengine.data as data
19+
import megengine.data.transform as T
20+
import megengine.distributed as dist
21+
import megengine.functional as F
22+
import megengine.optimizer as optim
2623

2724
logging = megengine.logger.get_logger()
2825

0 commit comments

Comments
 (0)