Merge pull request #9 from sovrasov/vs/tiny_fixes

pylint-related fixes

Author: sovrasov

annotation_converters/objectron_2_coco.py

@@ -30,8 +30,6 @@ def load_video_info(data_root, subset, classes):
                 avg_vid_len += len(ann)
                 vid_path = osp.join(data_root, 'videos' + osp.sep + line.strip() + osp.sep + 'video.MOV')
                 videos_info.append((vid_path, ann))
-                if len(videos_info) > 2:
-                    break
 
     avg_vid_len /= len(videos_info)
     return videos_info, avg_vid_len

annotation_converters/objectron_helpers.py

@@ -1,4 +1,3 @@
-from objectron.schema import annotation_data_pb2 as annotation_protocol
 import os
 import sys
 import subprocess
@@ -11,6 +10,9 @@
 if module_path not in sys.path:
     sys.path.append(module_path)
 
+#pylint: disable = wrong-import-position
+from objectron.schema import annotation_data_pb2 as annotation_protocol
+
 # The annotations are stored in protocol buffer format.
 # The AR Metadata captured with each frame in the video
 

tests/run_pylint.py

@@ -10,6 +10,7 @@
         '3rdparty/',
         'setup.py',
         'configs/detection',
+        '.history/',
         'torchdet3d/models'
     ]