mypy

Author: Matt Sokoloff

Makefile

@@ -6,7 +6,7 @@ test-staging: build
 	docker run -it -v ${PWD}:/usr/src -w /usr/src \
 		-e LABELBOX_TEST_ENVIRON="staging" \
 		-e LABELBOX_TEST_API_KEY_STAGING=${LABELBOX_TEST_API_KEY_STAGING} \
-		local/labelbox-python:test pytest $(PATH_TO_TEST) -svvx
+		local/labelbox-python:test pytest $(PATH_TO_TEST) -svv
 
 test-prod: build
 	docker run -it -v ${PWD}:/usr/src -w /usr/src \

labelbox/client.py

@@ -1,3 +1,4 @@
+# type: ignore
 from datetime import datetime, timezone
 import json
 import logging

labelbox/schema/data_row_metadata.py

@@ -1,5 +1,4 @@
 # type: ignore
-
 import datetime
 from enum import Enum
 from itertools import chain

mypy.ini

@@ -9,3 +9,6 @@ ignore_missing_imports = True
 
 [mypy-labelbox.data.*]
 ignore_errors = True
+
+[mypy-labelbox]
+ignore_errors = True

tests/integration/conftest.py

@@ -1,5 +1,7 @@
 import os
 import re
+import uuid
+import time
 from collections import namedtuple
 from datetime import datetime
 from enum import Enum
@@ -11,10 +13,13 @@
 
 from labelbox import Client
 from labelbox import LabelingFrontend
-from labelbox.orm.query import results_query_part
+from labelbox.orm import query
+from labelbox.schema.annotation_import import MALPredictionImport
+from labelbox.orm.db_object import Entity, DbObject
 from labelbox.pagination import PaginatedCollection
 from labelbox.schema.invite import Invite
 from labelbox.schema.user import User
+from labelbox import OntologyBuilder, Tool
 
 IMG_URL = "https://picsum.photos/200/300"
 
@@ -69,7 +74,7 @@ def get_project_invites(client, project_id):
         project(where: {id: $%s}) {id
         invites(from: $from, first: $first) { nodes { %s
         projectInvites { projectId projectRoleName } } nextCursor}}}
-    """ % (id_param, id_param, results_query_part(Invite))
+    """ % (id_param, id_param, query.results_query_part(Invite))
     return PaginatedCollection(client,
                                query_str, {id_param: project_id},
                                ['project', 'invites', 'nodes'],