recommended changes

Author: Matt Sokoloff

labelbox/data/serialization/labelbox_v1/converter.py

@@ -1,12 +1,12 @@
-from requests.exceptions import HTTPError
-import labelbox
 from typing import Any, Dict, Generator, Iterable
 import logging
 
 import ndjson
 import requests
+from requests.exceptions import HTTPError
 from google.api_core import retry
 
+import labelbox
 from .label import LBV1Label
 from ...annotation_types.collection import (LabelCollection, LabelGenerator,
                                             PrefetchGenerator)
@@ -17,12 +17,14 @@
 class LBV1Converter:
 
     @staticmethod
-    def deserialize_video(json_data: Iterable[Dict[str, Any]], client):
+    def deserialize_video(json_data: Iterable[Dict[str, Any]],
+                          client: labelbox.Client):
         """
         Converts a labelbox video export into the common labelbox format.
 
         Args:
             json_data: An iterable representing the labelbox video export.
+            client: The labelbox client for downloading video annotations
         Returns:
             LabelGenerator containing the video data.
         """
@@ -80,7 +82,7 @@ def _process(self, value):
         if 'frames' in value['Label']:
             req = self._request(value)
             value['Label'] = ndjson.loads(req)
-            return value
+        return value
 
     @retry.Retry(predicate=retry.if_exception_type(HTTPError))
     def _request(self, value):

labelbox/data/serialization/labelbox_v1/label.py

@@ -214,8 +214,8 @@ def _infer_media_type(self):
                 # To use this converter on data with this url format
                 #   filter out empty examples from the payload before deserializing.
                 raise TypeError(
-                    "Can't infer data type from row data. Remove empty examples before trying again"
-                )
+                    "Can't infer data type from row data. Remove empty examples before trying again. "
+                    f"row_data: {self.row_data[:200]}")
 
     def _has_object_annotations(self):
         return len(self.label.objects) > 0