dragoneye-python 0.4.1__tar.gz → 0.5.1__tar.gz

{dragoneye_python-0.4.1 → dragoneye_python-0.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragoneye-python
-Version: 0.4.1
+Version: 0.5.1
 License-Expression: MIT
 Requires-Python: >=3.8
 Requires-Dist: requests

{dragoneye_python-0.4.1 → dragoneye_python-0.5.1}/dragoneye_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragoneye-python
-Version: 0.4.1
+Version: 0.5.1
 License-Expression: MIT
 Requires-Python: >=3.8
 Requires-Dist: requests

{dragoneye_python-0.4.1 → dragoneye_python-0.5.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dragoneye-python"
-version = "0.4.1"
+version = "0.5.1"
 requires-python = ">=3.8"
 dynamic = ["dependencies"]
 license = "MIT"

dragoneye_python-0.5.1/src/dragoneye/__init__.py

@@ -0,0 +1,32 @@
+from .classification import (
+    Classification,
+)
+from .client import Dragoneye
+from .models import (
+    ClassificationAttributeOption,
+    ClassificationAttributeResponse,
+    ClassificationCategory,
+    ClassificationCategoryPrediction,
+    ClassificationObjectPrediction,
+    ClassificationPredictImageResponse,
+    ClassificationPredictVideoResponse,
+    ClassificationVideoObjectPrediction,
+)
+from .types.common import NormalizedBbox
+from .types.media import Image, Video
+
+__all__ = [
+    "Classification",
+    "ClassificationAttributeOption",
+    "ClassificationAttributeResponse",
+    "ClassificationCategory",
+    "ClassificationCategoryPrediction",
+    "ClassificationObjectPrediction",
+    "ClassificationPredictImageResponse",
+    "ClassificationPredictVideoResponse",
+    "ClassificationVideoObjectPrediction",
+    "Dragoneye",
+    "Image",
+    "NormalizedBbox",
+    "Video",
+]

{dragoneye_python-0.4.1 → dragoneye_python-0.5.1}/src/dragoneye/classification.py

@@ -255,7 +255,7 @@ class Classification:
         )
 
         predict_url = f"{BASE_API_URL}/predict"
-        predict_data = {
+        predict_data: dict[str, Any] = {
             "model_name": model_name,
             "prediction_task_uuid": prediction_task_begin_response.prediction_task_uuid,
             **kwargs,
@@ -354,6 +354,7 @@ class Classification:
 
         form_data = aiohttp.FormData()
         form_data.add_field("mimetype", mime_type)
+        form_data.add_field("response_version", "object")
         if file_name is not None:
             form_data.add_field("file_name", file_name)
 

{dragoneye_python-0.4.1 → dragoneye_python-0.5.1}/src/dragoneye/models.py

@@ -1,4 +1,4 @@
-from typing import Dict, Sequence
+from typing import Dict, List, Optional
 
 from pydantic import BaseModel
 
@@ -7,8 +7,6 @@ from dragoneye.types.common import (
     PredictionTaskState,
     PredictionTaskUUID,
     PredictionType,
-    TaxonID,
-    TaxonPrediction,
 )
 
 
@@ -18,22 +16,38 @@ class PredictionTaskStatusResponse(BaseModel):
     status: PredictionTaskState
 
 
-class ClassificationTraitRootPrediction(BaseModel):
-    id: TaxonID
+class ClassificationAttributeOption(BaseModel):
+    option_id: int
     name: str
-    displayName: str
-    taxons: Sequence[TaxonPrediction]
+    score: float
+
+
+class ClassificationAttributeResponse(BaseModel):
+    attribute_id: int
+    name: str
+    options: List[ClassificationAttributeOption]
+
+
+class ClassificationCategory(BaseModel):
+    id: int
+    name: str
+    score: float
+
+
+class ClassificationCategoryPrediction(BaseModel):
+    category: ClassificationCategory
+    attributes: List[ClassificationAttributeResponse]
 
 
 class ClassificationObjectPrediction(BaseModel):
     normalizedBbox: NormalizedBbox
-    category: TaxonPrediction
-    traits: Sequence[ClassificationTraitRootPrediction]
+    predictions: List[ClassificationCategoryPrediction]
 
 
 class ClassificationPredictImageResponse(BaseModel):
-    predictions: Sequence[ClassificationObjectPrediction]
+    object_predictions: List[ClassificationObjectPrediction]
     prediction_task_uuid: PredictionTaskUUID
+    original_file_name: Optional[str]
 
 
 class ClassificationVideoObjectPrediction(ClassificationObjectPrediction):
@@ -43,7 +57,8 @@ class ClassificationVideoObjectPrediction(ClassificationObjectPrediction):
 
 class ClassificationPredictVideoResponse(BaseModel):
     timestamp_us_to_predictions: Dict[
-        int, Sequence[ClassificationVideoObjectPrediction]
+        int, List[ClassificationVideoObjectPrediction]
     ]
     frames_per_second: int
     prediction_task_uuid: PredictionTaskUUID
+    original_file_name: Optional[str]

dragoneye_python-0.5.1/src/dragoneye/types/common.py

@@ -0,0 +1,12 @@
+from typing import Literal, NewType, Tuple
+
+PredictionType = Literal["image", "video"]
+PredictionTaskState = NewType("PredictionTaskState", str)
+
+NormalizedBbox = NewType("NormalizedBbox", Tuple[float, float, float, float])
+
+PredictionTaskUUID = NewType("PredictionTaskUUID", str)
+
+TimestampUs = NewType("TimestampUs", int)
+
+BASE_API_URL = "https://api.dragoneye.ai"

{dragoneye_python-0.4.1 → dragoneye_python-0.5.1}/src/dragoneye/types/media.py

@@ -136,10 +136,10 @@ class Media:
 
         # For any readable object with .read()
         if hasattr(src, "read"):
-            pos = _tell_safe(src)
-            data = src.read()
-            _seek_safe(src, pos)
-            return BytesIO(data)
+            pos = _tell_safe(src) # pyright: ignore [reportArgumentType]
+            data = src.read() # pyright: ignore [reportUnknownMemberType, reportAttributeAccessIssue, reportUnknownVariableType]
+            _seek_safe(src, pos) # pyright: ignore [reportArgumentType]
+            return BytesIO(data) # pyright: ignore [reportUnknownArgumentType]
 
         raise TypeError(
             "Invalid media source: expected bytes, BytesIO, or a readable binary stream."

dragoneye_python-0.4.1/src/dragoneye/__init__.py

@@ -1,25 +0,0 @@
-from .classification import (
-    Classification,
-)
-from .client import Dragoneye
-from .models import (
-    ClassificationObjectPrediction,
-    ClassificationPredictImageResponse,
-    ClassificationTraitRootPrediction,
-)
-from .types.common import NormalizedBbox, TaxonID, TaxonPrediction, TaxonType
-from .types.media import Image, Video
-
-__all__ = [
-    "Classification",
-    "ClassificationObjectPrediction",
-    "ClassificationPredictImageResponse",
-    "ClassificationTraitRootPrediction",
-    "Dragoneye",
-    "Image",
-    "Video",
-    "NormalizedBbox",
-    "TaxonID",
-    "TaxonPrediction",
-    "TaxonType",
-]

dragoneye_python-0.4.1/src/dragoneye/types/common.py

@@ -1,31 +0,0 @@
-from enum import Enum
-from typing import Literal, NewType, Optional, Sequence, Tuple
-
-from pydantic import BaseModel
-
-PredictionType = Literal["image", "video"]
-PredictionTaskState = NewType("PredictionTaskState", str)
-
-NormalizedBbox = NewType("NormalizedBbox", Tuple[float, float, float, float])
-
-
-class TaxonType(str, Enum):
-    CATEGORY = ("category",)
-    TRAIT = ("trait",)
-
-
-TaxonID = NewType("TaxonID", int)
-
-PredictionTaskUUID = NewType("PredictionTaskUUID", str)
-
-
-class TaxonPrediction(BaseModel):
-    id: TaxonID
-    type: TaxonType
-    name: str
-    displayName: str
-    score: Optional[float]
-    children: Sequence["TaxonPrediction"]
-
-
-BASE_API_URL = "https://api.dragoneye.ai"