Skip to content

Commit ea893d3

Browse files
committed
Fix lint
1 parent 8ad6930 commit ea893d3

File tree

3 files changed

+3
-2
lines changed

3 files changed

+3
-2
lines changed

torchvision/csrc/ROIPool.h

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,6 @@ at::Tensor ROIPool_backward(
6565
width);
6666
}
6767

68-
6968
using namespace at;
7069
using torch::Tensor;
7170
using torch::autograd::AutogradContext;

torchvision/csrc/vision.cpp

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,6 @@ PyMODINIT_FUNC PyInit__custom_ops(void) {
2727
#endif
2828
#endif
2929

30-
3130
int64_t _cuda_version() {
3231
#ifdef WITH_CUDA
3332
return CUDA_VERSION;

torchvision/extension.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
_HAS_OPS = False
22

3+
34
def _register_extensions():
45
import os
56
import imp
@@ -10,6 +11,7 @@ def _register_extensions():
1011
_, path, _ = imp.find_module("_C", [lib_dir])
1112
torch.ops.load_library(path)
1213

14+
1315
try:
1416
_register_extensions()
1517
_HAS_OPS = True
@@ -44,4 +46,5 @@ def _check_cuda_version():
4446
.format(t_major, t_minor, tv_major, tv_minor))
4547
return _version
4648

49+
4750
_check_cuda_version()

0 commit comments

Comments
 (0)