Update HTTP annotation response router.

vision/google/cloud/vision/annotations.py

@@ -14,6 +14,7 @@
 
 """Annotations management for Vision API responses."""
 
+import six
 
 from google.cloud.vision.color import ImagePropertiesAnnotation
 from google.cloud.vision.entity import EntityAnnotation
@@ -86,14 +87,11 @@ def from_api_repr(cls, response):
         :rtype: :class:`~google.cloud.vision.annotations.Annotations`
         :returns: An instance of ``Annotations`` with detection types loaded.
         """
-        annotations = {}
-        for feature_type, annotation in response.items():
-            curr_feature = annotations.setdefault(_KEY_MAP[feature_type], [])
-            entity = _entity_from_response_type(feature_type, annotation)
-            if isinstance(entity, list):
-                curr_feature.extend(entity)
-            else:
-                annotations[_KEY_MAP[feature_type]] = entity
+        annotations = {
+            _KEY_MAP[feature_type]: _entity_from_response_type(
+                feature_type, annotation)
+            for feature_type, annotation in six.iteritems(response)
+        }
         return cls(**annotations)
 
     @classmethod