intellif-aihub 0.1.3__py3-none-any.whl → 0.1.5__py3-none-any.whl

aihub/models/dataset_management.py

@@ -1,99 +1,99 @@
-from __future__ import annotations
-
-from typing import List, Optional
-
-from pydantic import BaseModel, Field
-
-
-class CreateDatasetRequest(BaseModel):
-    name: str
-    description: str
-    tags: List[int]
-    cover_img: Optional[str] = Field(None, alias="cover_img")
-    create_by: Optional[int] = Field(None, alias="create_by")
-    is_private: Optional[bool] = Field(None, alias="is_private")
-    access_user_ids: Optional[List[int]] = Field(None, alias="access_user_ids")
-
-
-class CreateDatasetResponse(BaseModel):
-    id: int = Field(alias="id")
-
-
-class DatasetVersionBase(BaseModel):
-    id: int
-    version: int
-    status: int
-    parquet_index_path: Optional[str] = Field(None, alias="parquet_index_path")
-    data_count: int = Field(alias="data_count")
-
-
-class DatasetDetail(BaseModel):
-    id: int
-    name: str
-    description: str
-    cover_img: Optional[str] = Field(None, alias="cover_img")
-    created_at: int = Field(alias="created_at")
-    updated_at: int = Field(alias="update_at")
-    user_id: int = Field(alias="user_id")
-    username: str
-    tags: List[int]
-    access_user_ids: Optional[List[int]] = Field(None, alias="access_user_ids")
-    is_private: Optional[bool] = Field(None, alias="is_private")
-    versions: List[DatasetVersionBase]
-
-
-class ExtInfo(BaseModel):
-    rec_file_path: Optional[str] = Field(None, alias="rec_file_path")
-    idx_file_path: Optional[str] = Field(None, alias="idx_file_path")
-    json_file_path: Optional[str] = Field(None, alias="json_file_path")
-    image_dir_path: Optional[str] = Field(None, alias="image_dir_path")
-
-
-class CreateDatasetVersionRequest(BaseModel):
-    upload_path: str = Field(alias="upload_path")
-    description: Optional[str] = None
-    dataset_id: int = Field(alias="dataset_id")
-    object_cnt: Optional[int] = Field(None, alias="object_cnt")
-    data_size: Optional[int] = Field(None, alias="data_size")
-    create_by: Optional[int] = Field(None, alias="create_by")
-    upload_type: Optional[int] = Field(4, alias="upload_type")
-    ext_info: Optional[ExtInfo] = Field(None, alias="ext_info")
-
-
-class CreateDatasetVersionResponse(BaseModel):
-    id: int = Field(alias="id")
-
-
-class UploadDatasetVersionRequest(BaseModel):
-    upload_path: str = Field(alias="upload_path")
-    upload_type: int = Field(alias="upload_type")
-    dataset_id: int = Field(alias="dataset_id")
-    parent_version_id: Optional[int] = Field(None, alias="parent_version_id")
-    description: Optional[str] = Field(None, alias="description")
-
-
-class UploadDatasetVersionResponse(BaseModel):
-    id: int = Field(alias="id")
-
-
-class DatasetVersionDetail(BaseModel):
-    id: int
-    version: int
-    dataset_id: int = Field(alias="dataset_id")
-    upload_path: str = Field(alias="upload_path")
-    upload_type: int = Field(alias="upload_type")
-    parent_version_id: Optional[int] = Field(None, alias="parent_version_id")
-    description: Optional[str] = None
-    status: int
-    message: Optional[str] = None
-    created_at: int = Field(alias="created_at")
-    user_id: int = Field(alias="user_id")
-    data_size: Optional[int] = Field(None, alias="data_size")
-    data_count: Optional[int] = Field(None, alias="data_count")
-    parquet_index_path: Optional[str] = Field(None, alias="parquet_index_path")
-    ext_info: Optional[ExtInfo] = Field(None, alias="ext_info")
-
-
-class FileUploadData(BaseModel):
-    path: str
-    url: str
+from __future__ import annotations
+
+from typing import List, Optional
+
+from pydantic import BaseModel, Field
+
+
+class CreateDatasetRequest(BaseModel):
+    name: str
+    description: str
+    tags: List[int]
+    cover_img: Optional[str] = Field(None, alias="cover_img")
+    create_by: Optional[int] = Field(None, alias="create_by")
+    is_private: Optional[bool] = Field(None, alias="is_private")
+    access_user_ids: Optional[List[int]] = Field(None, alias="access_user_ids")
+
+
+class CreateDatasetResponse(BaseModel):
+    id: int = Field(alias="id")
+
+
+class DatasetVersionBase(BaseModel):
+    id: int
+    version: int
+    status: int
+    parquet_index_path: Optional[str] = Field(None, alias="parquet_index_path")
+    data_count: int = Field(alias="data_count")
+
+
+class DatasetDetail(BaseModel):
+    id: int
+    name: str
+    description: str
+    cover_img: Optional[str] = Field(None, alias="cover_img")
+    created_at: int = Field(alias="created_at")
+    updated_at: int = Field(alias="update_at")
+    user_id: int = Field(alias="user_id")
+    username: str
+    tags: List[int]
+    access_user_ids: Optional[List[int]] = Field(None, alias="access_user_ids")
+    is_private: Optional[bool] = Field(None, alias="is_private")
+    versions: List[DatasetVersionBase]
+
+
+class ExtInfo(BaseModel):
+    rec_file_path: Optional[str] = Field(None, alias="rec_file_path")
+    idx_file_path: Optional[str] = Field(None, alias="idx_file_path")
+    json_file_path: Optional[str] = Field(None, alias="json_file_path")
+    image_dir_path: Optional[str] = Field(None, alias="image_dir_path")
+
+
+class CreateDatasetVersionRequest(BaseModel):
+    upload_path: str = Field(alias="upload_path")
+    description: Optional[str] = None
+    dataset_id: int = Field(alias="dataset_id")
+    object_cnt: Optional[int] = Field(None, alias="object_cnt")
+    data_size: Optional[int] = Field(None, alias="data_size")
+    create_by: Optional[int] = Field(None, alias="create_by")
+    upload_type: Optional[int] = Field(4, alias="upload_type")
+    ext_info: Optional[ExtInfo] = Field(None, alias="ext_info")
+
+
+class CreateDatasetVersionResponse(BaseModel):
+    id: int = Field(alias="id")
+
+
+class UploadDatasetVersionRequest(BaseModel):
+    upload_path: str = Field(alias="upload_path")
+    upload_type: int = Field(alias="upload_type")
+    dataset_id: int = Field(alias="dataset_id")
+    parent_version_id: Optional[int] = Field(None, alias="parent_version_id")
+    description: Optional[str] = Field(None, alias="description")
+
+
+class UploadDatasetVersionResponse(BaseModel):
+    id: int = Field(alias="id")
+
+
+class DatasetVersionDetail(BaseModel):
+    id: int
+    version: int
+    dataset_id: int = Field(alias="dataset_id")
+    upload_path: str = Field(alias="upload_path")
+    upload_type: int = Field(alias="upload_type")
+    parent_version_id: Optional[int] = Field(None, alias="parent_version_id")
+    description: Optional[str] = None
+    status: int
+    message: Optional[str] = None
+    created_at: int = Field(alias="created_at")
+    user_id: int = Field(alias="user_id")
+    data_size: Optional[int] = Field(None, alias="data_size")
+    data_count: Optional[int] = Field(None, alias="data_count")
+    parquet_index_path: Optional[str] = Field(None, alias="parquet_index_path")
+    ext_info: Optional[ExtInfo] = Field(None, alias="ext_info")
+
+
+class FileUploadData(BaseModel):
+    path: str
+    url: str

aihub/models/document_center.py

@@ -1,28 +1,28 @@
-# !/usr/bin/env python
-# -*-coding:utf-8 -*-
-
-from __future__ import annotations
-
-from typing import List, Optional, Any
-
-from pydantic import BaseModel
-
-
-class Document(BaseModel):
-    id: int
-    title: str
-    type: int
-    edit_time: int
-    need_update: bool
-    content: str
-    username: str
-    user_id: int
-    created_at: int
-    comments: Optional[Any] = None
-
-
-class GetDocumentsResponse(BaseModel):
-    total: int
-    page_size: int
-    page_num: int
-    data: List[Document]
+# !/usr/bin/env python
+# -*-coding:utf-8 -*-
+
+from __future__ import annotations
+
+from typing import List, Optional, Any
+
+from pydantic import BaseModel
+
+
+class Document(BaseModel):
+    id: int
+    title: str
+    type: int
+    edit_time: int
+    need_update: bool
+    content: str
+    username: str
+    user_id: int
+    created_at: int
+    comments: Optional[Any] = None
+
+
+class GetDocumentsResponse(BaseModel):
+    total: int
+    page_size: int
+    page_num: int
+    data: List[Document]

aihub/models/eval.py

@@ -0,0 +1,17 @@
+# !/usr/bin/env python
+# -*-coding:utf-8 -*-
+from pydantic import BaseModel
+
+
+class CreatEvalReq(BaseModel):
+    dataset_id: int
+    dataset_version_id: int
+    prediction_artifact_path: str
+    evaled_artifact_path: str
+    run_id: str
+    user_id: int = 0
+    report: dict = {}
+
+
+class CreatEvalResp(BaseModel):
+    id: int

aihub/models/labelfree.py

@@ -1,31 +1,53 @@
-from __future__ import annotations
-
-from typing import Optional
-
-from pydantic import BaseModel, Field
-
-
-class Stats(BaseModel):
-    """标注统计信息"""
-
-    total_annotations: int = Field(alias="total_annotations")
-    labeled_annotations: int = Field(alias="labeled_annotations")
-    total_labels: int = Field(alias="total_labels")
-    total_reviews: Optional[int] = Field(None, alias="total_reviews")
-    unlabeled_reviews: Optional[int] = Field(None, alias="unlabeled_reviews")
-    labeled_reviews: Optional[int] = Field(None, alias="labeled_reviews")
-    accepted_count: Optional[int] = Field(None, alias="accepted_count")
-    rejected_count: Optional[int] = Field(None, alias="rejected_count")
-
-
-class GetGlobalStatsResponse(BaseModel):
-    """
-    标注统计概况
-    """
-
-    global_stats: Stats = Field(alias="global_stats")
-    valid_ten_percent: bool = Field(alias="valid_ten_percent")
-    valid_fifty_percent: bool = Field(alias="valid_fifty_percent")
-    valid_hundred_percent: bool = Field(alias="valid_hundred_percent")
-    data_exported_count: int = Field(alias="data_exported_count")
-    exported_dataset_name: str = Field(alias="exported_dataset_name")
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import BaseModel, Field
+
+
+class Stats(BaseModel):
+    """标注统计信息"""
+
+    total_annotations: int = Field(alias="total_annotations", description="总数据量")
+    labeled_annotations: int = Field(
+        alias="labeled_annotations", description="已标注数据量"
+    )
+    total_labels: int = Field(alias="total_labels", description="总标签量")
+    total_reviews: Optional[int] = Field(
+        None, alias="total_reviews", description="总质检量"
+    )
+    unlabeled_reviews: Optional[int] = Field(
+        None, alias="unlabeled_reviews", description="未质检量"
+    )
+    labeled_reviews: Optional[int] = Field(
+        None, alias="labeled_reviews", description="已质检量"
+    )
+    accepted_count: Optional[int] = Field(
+        None, alias="accepted_count", description="质检通过量"
+    )
+    rejected_count: Optional[int] = Field(
+        None, alias="rejected_count", description="质检打回量"
+    )
+
+
+class GetGlobalStatsResponse(BaseModel):
+    """
+    标注统计概况
+    """
+
+    global_stats: Stats = Field(alias="global_stats")
+    valid_ten_percent: bool = Field(
+        alias="valid_ten_percent", description="是否完成验收10%"
+    )
+    valid_fifty_percent: bool = Field(
+        alias="valid_fifty_percent", description="是否完成验收50%"
+    )
+    valid_hundred_percent: bool = Field(
+        alias="valid_hundred_percent", description="是否完成验收100%"
+    )
+    data_exported_count: int = Field(
+        alias="data_exported_count", description="已导出数据次数"
+    )
+    exported_dataset_name: str = Field(
+        alias="exported_dataset_name", description="最新数据集名称"
+    )