denkproto 1.0.91__py3-none-any.whl → 1.0.93__py3-none-any.whl

denkproto/json/prediction_request.py

@@ -0,0 +1,263 @@
+from uuid import UUID
+from typing import Any, Dict, List, TypeVar, Callable, Type, cast
+
+
+T = TypeVar("T")
+
+
+def from_int(x: Any) -> int:
+    assert isinstance(x, int) and not isinstance(x, bool)
+    return x
+
+
+def from_dict(f: Callable[[Any], T], x: Any) -> Dict[str, T]:
+    assert isinstance(x, dict)
+    return { k: f(v) for (k, v) in x.items() }
+
+
+def from_bool(x: Any) -> bool:
+    assert isinstance(x, bool)
+    return x
+
+
+def to_class(c: Type[T], x: Any) -> dict:
+    assert isinstance(x, c)
+    return cast(Any, x).to_dict()
+
+
+def from_list(f: Callable[[Any], T], x: Any) -> List[T]:
+    assert isinstance(x, list)
+    return [f(y) for y in x]
+
+
+def from_str(x: Any) -> str:
+    assert isinstance(x, str)
+    return x
+
+
+class Image:
+    blob_id: UUID
+    height: int
+    owned_by_group_id: UUID
+    width: int
+
+    def __init__(self, blob_id: UUID, height: int, owned_by_group_id: UUID, width: int) -> None:
+        self.blob_id = blob_id
+        self.height = height
+        self.owned_by_group_id = owned_by_group_id
+        self.width = width
+
+    @staticmethod
+    def from_dict(obj: Any) -> 'Image':
+        assert isinstance(obj, dict)
+        blob_id = UUID(obj.get("blob_id"))
+        height = from_int(obj.get("height"))
+        owned_by_group_id = UUID(obj.get("owned_by_group_id"))
+        width = from_int(obj.get("width"))
+        return Image(blob_id, height, owned_by_group_id, width)
+
+    def to_dict(self) -> dict:
+        result: dict = {}
+        result["blob_id"] = str(self.blob_id)
+        result["height"] = from_int(self.height)
+        result["owned_by_group_id"] = str(self.owned_by_group_id)
+        result["width"] = from_int(self.width)
+        return result
+
+
+class ClassLabel:
+    id: UUID
+    idx: int
+
+    def __init__(self, id: UUID, idx: int) -> None:
+        self.id = id
+        self.idx = idx
+
+    @staticmethod
+    def from_dict(obj: Any) -> 'ClassLabel':
+        assert isinstance(obj, dict)
+        id = UUID(obj.get("id"))
+        idx = from_int(obj.get("idx"))
+        return ClassLabel(id, idx)
+
+    def to_dict(self) -> dict:
+        result: dict = {}
+        result["id"] = str(self.id)
+        result["idx"] = from_int(self.idx)
+        return result
+
+
+class Config:
+    metadata: Dict[str, Any]
+    uses_validation_tiling: bool
+
+    def __init__(self, metadata: Dict[str, Any], uses_validation_tiling: bool) -> None:
+        self.metadata = metadata
+        self.uses_validation_tiling = uses_validation_tiling
+
+    @staticmethod
+    def from_dict(obj: Any) -> 'Config':
+        assert isinstance(obj, dict)
+        metadata = from_dict(lambda x: x, obj.get("metadata"))
+        uses_validation_tiling = from_bool(obj.get("uses_validation_tiling"))
+        return Config(metadata, uses_validation_tiling)
+
+    def to_dict(self) -> dict:
+        result: dict = {}
+        result["metadata"] = from_dict(lambda x: x, self.metadata)
+        result["uses_validation_tiling"] = from_bool(self.uses_validation_tiling)
+        return result
+
+
+class Onnx:
+    blob_id: UUID
+    owned_by_group_id: UUID
+
+    def __init__(self, blob_id: UUID, owned_by_group_id: UUID) -> None:
+        self.blob_id = blob_id
+        self.owned_by_group_id = owned_by_group_id
+
+    @staticmethod
+    def from_dict(obj: Any) -> 'Onnx':
+        assert isinstance(obj, dict)
+        blob_id = UUID(obj.get("blob_id"))
+        owned_by_group_id = UUID(obj.get("owned_by_group_id"))
+        return Onnx(blob_id, owned_by_group_id)
+
+    def to_dict(self) -> dict:
+        result: dict = {}
+        result["blob_id"] = str(self.blob_id)
+        result["owned_by_group_id"] = str(self.owned_by_group_id)
+        return result
+
+
+class Pytorch:
+    blob_id: UUID
+    owned_by_group_id: UUID
+
+    def __init__(self, blob_id: UUID, owned_by_group_id: UUID) -> None:
+        self.blob_id = blob_id
+        self.owned_by_group_id = owned_by_group_id
+
+    @staticmethod
+    def from_dict(obj: Any) -> 'Pytorch':
+        assert isinstance(obj, dict)
+        blob_id = UUID(obj.get("blob_id"))
+        owned_by_group_id = UUID(obj.get("owned_by_group_id"))
+        return Pytorch(blob_id, owned_by_group_id)
+
+    def to_dict(self) -> dict:
+        result: dict = {}
+        result["blob_id"] = str(self.blob_id)
+        result["owned_by_group_id"] = str(self.owned_by_group_id)
+        return result
+
+
+class Snapshot:
+    onnx: Onnx
+    pytorch: Pytorch
+
+    def __init__(self, onnx: Onnx, pytorch: Pytorch) -> None:
+        self.onnx = onnx
+        self.pytorch = pytorch
+
+    @staticmethod
+    def from_dict(obj: Any) -> 'Snapshot':
+        assert isinstance(obj, dict)
+        onnx = Onnx.from_dict(obj.get("onnx"))
+        pytorch = Pytorch.from_dict(obj.get("pytorch"))
+        return Snapshot(onnx, pytorch)
+
+    def to_dict(self) -> dict:
+        result: dict = {}
+        result["onnx"] = to_class(Onnx, self.onnx)
+        result["pytorch"] = to_class(Pytorch, self.pytorch)
+        return result
+
+
+class NetworkExperiment:
+    class_labels: List[ClassLabel]
+    config: Config
+    flavor: str
+    network_typename: str
+    snapshot: Snapshot
+
+    def __init__(self, class_labels: List[ClassLabel], config: Config, flavor: str, network_typename: str, snapshot: Snapshot) -> None:
+        self.class_labels = class_labels
+        self.config = config
+        self.flavor = flavor
+        self.network_typename = network_typename
+        self.snapshot = snapshot
+
+    @staticmethod
+    def from_dict(obj: Any) -> 'NetworkExperiment':
+        assert isinstance(obj, dict)
+        class_labels = from_list(ClassLabel.from_dict, obj.get("class_labels"))
+        config = Config.from_dict(obj.get("config"))
+        flavor = from_str(obj.get("flavor"))
+        network_typename = from_str(obj.get("network_typename"))
+        snapshot = Snapshot.from_dict(obj.get("snapshot"))
+        return NetworkExperiment(class_labels, config, flavor, network_typename, snapshot)
+
+    def to_dict(self) -> dict:
+        result: dict = {}
+        result["class_labels"] = from_list(lambda x: to_class(ClassLabel, x), self.class_labels)
+        result["config"] = to_class(Config, self.config)
+        result["flavor"] = from_str(self.flavor)
+        result["network_typename"] = from_str(self.network_typename)
+        result["snapshot"] = to_class(Snapshot, self.snapshot)
+        return result
+
+
+class PredictionRequest:
+    created_by_user_id: UUID
+    hasura_url: str
+    id: UUID
+    image: Image
+    network_experiment: NetworkExperiment
+    owned_by_group_id: UUID
+    prediction_priority: int
+    request_classification_interpretation: bool
+
+    def __init__(self, created_by_user_id: UUID, hasura_url: str, id: UUID, image: Image, network_experiment: NetworkExperiment, owned_by_group_id: UUID, prediction_priority: int, request_classification_interpretation: bool) -> None:
+        self.created_by_user_id = created_by_user_id
+        self.hasura_url = hasura_url
+        self.id = id
+        self.image = image
+        self.network_experiment = network_experiment
+        self.owned_by_group_id = owned_by_group_id
+        self.prediction_priority = prediction_priority
+        self.request_classification_interpretation = request_classification_interpretation
+
+    @staticmethod
+    def from_dict(obj: Any) -> 'PredictionRequest':
+        assert isinstance(obj, dict)
+        created_by_user_id = UUID(obj.get("created_by_user_id"))
+        hasura_url = from_str(obj.get("hasura_url"))
+        id = UUID(obj.get("id"))
+        image = Image.from_dict(obj.get("image"))
+        network_experiment = NetworkExperiment.from_dict(obj.get("network_experiment"))
+        owned_by_group_id = UUID(obj.get("owned_by_group_id"))
+        prediction_priority = from_int(obj.get("prediction_priority"))
+        request_classification_interpretation = from_bool(obj.get("request_classification_interpretation"))
+        return PredictionRequest(created_by_user_id, hasura_url, id, image, network_experiment, owned_by_group_id, prediction_priority, request_classification_interpretation)
+
+    def to_dict(self) -> dict:
+        result: dict = {}
+        result["created_by_user_id"] = str(self.created_by_user_id)
+        result["hasura_url"] = from_str(self.hasura_url)
+        result["id"] = str(self.id)
+        result["image"] = to_class(Image, self.image)
+        result["network_experiment"] = to_class(NetworkExperiment, self.network_experiment)
+        result["owned_by_group_id"] = str(self.owned_by_group_id)
+        result["prediction_priority"] = from_int(self.prediction_priority)
+        result["request_classification_interpretation"] = from_bool(self.request_classification_interpretation)
+        return result
+
+
+def prediction_request_from_dict(s: Any) -> PredictionRequest:
+    return PredictionRequest.from_dict(s)
+
+
+def prediction_request_to_dict(x: PredictionRequest) -> Any:
+    return to_class(PredictionRequest, x)

{denkproto-1.0.91.dist-info → denkproto-1.0.93.dist-info}/METADATA

@@ -1,5 +1,5 @@
 Metadata-Version: 2.3
 Name: denkproto
-Version: 1.0.91
+Version: 1.0.93
 Requires-Python: >=3.10
 Requires-Dist: protobuf>=3.20.3

{denkproto-1.0.91.dist-info → denkproto-1.0.93.dist-info}/RECORD

@@ -4,7 +4,7 @@ denkproto/DENKbuffer_pb2_grpc.py,sha256=-CPJPM4FOqwvwV8-f1iJlD18UD9juVIIHfdWUecu
 denkproto/ImageAnalysis_ProtobufMessages_pb2.py,sha256=iEY0j9ySGUThnqTdYD4uAVr9P3GiC5R02iK53zEOXUQ,21015
 denkproto/ImageAnalysis_ProtobufMessages_pb2.pyi,sha256=5LFtxrmYpJHizDDNGFTkL7-NQ_TkwqCSdq7vcv3lg-c,36243
 denkproto/ImageAnalysis_ProtobufMessages_pb2_grpc.py,sha256=l3agtDjgu4jay6P9TRnHhyhJ-7UdoII27ywhw3k84oo,911
-denkproto/__about__.py,sha256=B1iOLIpXYedqoLNGFLT8CrCSzMV5ZMTfScfMNbJ4nNE,23
+denkproto/__about__.py,sha256=BTvw8DmB4FrGsu9dg34XOAthT2XrdjaaYC_7bKzDlE4,23
 denkproto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 denkproto/denkcache_pb2.py,sha256=u0O26m7t4kfu4R1nx1ZcTst4n6pG32pMbhl2PGYivXE,7161
 denkproto/denkcache_pb2.pyi,sha256=8K_Ebyy4mgXrxqJenN8f8LXLvVKOiaZxhmGeYjFZVpY,6357
@@ -26,11 +26,14 @@ denkproto/validate_pb2.py,sha256=CuGAaHir9X9jniW3QsRKAESjYzoS2U6dLk_J55XmNqU,136
 denkproto/validate_pb2.pyi,sha256=fWsdVOR3NJDioCKkCKfxfl4qaEb5xqXXU_WlEbdQx6E,23077
 denkproto/validate_pb2_grpc.py,sha256=XvjuWEgJFJtH1E7HWm7SKpV10PMpOSbonKa2VPHpYy8,889
 denkproto/json/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+denkproto/json/annotation_comparer_request.py,sha256=2vUi2IZaRLF_oqNgGTyOxw0aFaLHMPikocnvjmJ5KMs,39633
 denkproto/json/classification_markup.py,sha256=vTu0H7Cb3gU6UUUSg1vDTRlFUorZrjMbcp_yx6UssZA,2461
 denkproto/json/inference_graph_models_generated.py,sha256=GfsGk4bKbL8gyqOScA4x13P8sJrAmv3tEhNU36rbIDs,6313
 denkproto/json/object_detection_markup.py,sha256=T0hcFPq8F_galjDjRC9dbcRVwCSOYtu2jt9wpEeHlQs,4904
 denkproto/json/ocr_markup.py,sha256=KyOpth9evOekyhTJdZSnYyB9EIyoWbY33sqncb_jBgw,7069
+denkproto/json/ocr_prediction_request.py,sha256=PXY_1hmPQcKpHHZyikMFG_S0HKiwbXtRv5oW5x1zHRM,17783
+denkproto/json/prediction_request.py,sha256=N7KZaKr9FW8RGme86O9xb2JjNBaJDv1TW2ffCugckj0,8839
 denkproto/json/segmentation_markup.py,sha256=EvniRksF2KaQolm6zZ6UKSiGwnqc8wR2sHB1iv05RTE,19911
-denkproto-1.0.91.dist-info/METADATA,sha256=pvRt_IA1uBvij9Yke6Hqqsh-la0rhs1bvO-GHE67g_I,110
-denkproto-1.0.91.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-denkproto-1.0.91.dist-info/RECORD,,
+denkproto-1.0.93.dist-info/METADATA,sha256=o__s_qTRs9Iqj-Dg0AsZLF0jHHoiAndRHNKtBdNRwSw,110
+denkproto-1.0.93.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+denkproto-1.0.93.dist-info/RECORD,,