aignostics 0.2.205__tar.gz → 0.2.207__tar.gz

{aignostics-0.2.205 → aignostics-0.2.207}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aignostics
-Version: 0.2.205
+Version: 0.2.207
 Summary: 🔬 Python SDK providing access to the Aignostics Platform. Includes Aignostics Launchpad (Desktop Application), Aignostics CLI (Command-Line Interface), example notebooks, and Aignostics Client Library.
 Project-URL: Homepage, https://aignostics.readthedocs.io/en/latest/
 Project-URL: Documentation, https://aignostics.readthedocs.io/en/latest/

{aignostics-0.2.205 → aignostics-0.2.207}/examples/notebook.py

@@ -2,7 +2,7 @@
 # requires-python = ">=3.13"
 # dependencies = [
 #     "marimo",
-#     "aignostics==0.2.205",
+#     "aignostics==0.2.207",
 # ]
 # ///
 

{aignostics-0.2.205 → aignostics-0.2.207}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "aignostics"
-version = "0.2.205"
+version = "0.2.207"
 description = "🔬 Python SDK providing access to the Aignostics Platform. Includes Aignostics Launchpad (Desktop Application), Aignostics CLI (Command-Line Interface), example notebooks, and Aignostics Client Library."
 readme = "README.md"
 authors = [
@@ -372,7 +372,7 @@ source = ["src/"]
 
 
 [tool.bumpversion]
-current_version = "0.2.205"
+current_version = "0.2.207"
 parse = "(?P<major>\\d+)\\.(?P<minor>\\d+)\\.(?P<patch>\\d+)"
 serialize = ["{major}.{minor}.{patch}"]
 search = "{current_version}"

{aignostics-0.2.205 → aignostics-0.2.207}/src/aignostics/application/_cli.py

@@ -746,31 +746,22 @@ def run_submit(  # noqa: PLR0913, PLR0917
             metadata_dict,
         )
 
-        # Build custom metadata with pipeline configuration
-        custom_metadata = {
-            "pipeline": {
-                "gpu": {
-                    "gpu_type": gpu_type,
-                    "provisioning_mode": gpu_provisioning_mode,
-                    "max_gpus_per_slide": max_gpus_per_slide,
-                },
-                "cpu": {
-                    "provisioning_mode": cpu_provisioning_mode,
-                },
-            },
-        }
-
+        # Submit run with pipeline configuration
         application_run = Service().application_run_submit_from_metadata(
             application_id=application_id,
             metadata=metadata_dict,
             application_version=application_version,
-            custom_metadata=custom_metadata,
+            custom_metadata=None,
             note=note,
             tags={tag.strip() for tag in tags.split(",") if tag.strip()} if tags else None,
             due_date=due_date,
             deadline=deadline,
             onboard_to_aignostics_portal=onboard_to_aignostics_portal,
             validate_only=validate_only,
+            gpu_type=gpu_type,
+            gpu_provisioning_mode=gpu_provisioning_mode,
+            max_gpus_per_slide=max_gpus_per_slide,
+            cpu_provisioning_mode=cpu_provisioning_mode,
         )
         console.print(
             f"Submitted run with id '{application_run.run_id}' for "

{aignostics-0.2.205 → aignostics-0.2.207}/src/aignostics/application/_gui/_frame.py

@@ -11,7 +11,7 @@ from .._service import Service  # noqa: TID252
 from ._utils import application_id_to_icon, run_status_to_icon_and_color
 
 BORDERED_SEPARATOR = "bordered separator"
-RUNS_LIMIT = 500
+RUNS_LIMIT = 200
 RUNS_REFRESH_INTERVAL = 60 * 15  # 15 minutes
 STORAGE_TAB_RUNS_HAS_OUTPUT = "runs_has_output"
 

{aignostics-0.2.205 → aignostics-0.2.207}/src/aignostics/application/_gui/_page_application_describe.py

@@ -715,25 +715,12 @@ async def _page_application_describe(application_id: str) -> None:  # noqa: C901
             """Submit the application run."""
             ui.notify("Submitting application run ...", type="info")
             try:
-                # Build custom metadata with pipeline configuration
-                custom_metadata = {
-                    "pipeline": {
-                        "gpu": {
-                            "gpu_type": submit_form.gpu_type,
-                            "provisioning_mode": submit_form.gpu_provisioning_mode,
-                            "max_gpus_per_slide": submit_form.max_gpus_per_slide,
-                        },
-                        "cpu": {
-                            "provisioning_mode": submit_form.cpu_provisioning_mode,
-                        },
-                    },
-                }
-
+                # Submit run with pipeline configuration
                 run = service.application_run_submit_from_metadata(
                     application_id=str(submit_form.application_id),
                     metadata=submit_form.metadata or [],
                     application_version=str(submit_form.application_version),
-                    custom_metadata=custom_metadata,
+                    custom_metadata=None,
                     note=submit_form.note,
                     tags=set(submit_form.tags) if submit_form.tags else None,
                     due_date=datetime.strptime(submit_form.due_date, "%Y-%m-%d %H:%M")
@@ -746,6 +733,10 @@ async def _page_application_describe(application_id: str) -> None:  # noqa: C901
                     .isoformat(),
                     validate_only=submit_form.validate_only,
                     onboard_to_aignostics_portal=submit_form.onboard_to_aignostics_portal,
+                    gpu_type=submit_form.gpu_type,
+                    gpu_provisioning_mode=submit_form.gpu_provisioning_mode,
+                    max_gpus_per_slide=submit_form.max_gpus_per_slide,
+                    cpu_provisioning_mode=submit_form.cpu_provisioning_mode,
                 )
             except Exception as e:
                 ui.notify(

{aignostics-0.2.205 → aignostics-0.2.207}/src/aignostics/application/_service.py

@@ -802,6 +802,10 @@ class Service(BaseService):  # noqa: PLR0904
         deadline: str | None = None,
         onboard_to_aignostics_portal: bool = False,
         validate_only: bool = False,
+        gpu_type: str | None = None,
+        gpu_provisioning_mode: str | None = None,
+        max_gpus_per_slide: int | None = None,
+        cpu_provisioning_mode: str | None = None,
     ) -> Run:
         """Submit a run for the given application.
 
@@ -820,6 +824,10 @@ class Service(BaseService):  # noqa: PLR0904
                 If not given latest version is used.
             onboard_to_aignostics_portal (bool): True if the run should be onboarded to the Aignostics Portal.
             validate_only (bool): If True, cancel the run post validation, before analysis.
+            gpu_type (str | None): The type of GPU to use (L4 or A100).
+            gpu_provisioning_mode (str | None): The provisioning mode for GPU resources (SPOT or ON_DEMAND).
+            max_gpus_per_slide (int | None): The maximum number of GPUs to allocate per slide.
+            cpu_provisioning_mode (str | None): The provisioning mode for CPU resources (SPOT or ON_DEMAND).
 
         Returns:
             Run: The submitted run.
@@ -918,6 +926,10 @@ class Service(BaseService):  # noqa: PLR0904
                 deadline=deadline,
                 onboard_to_aignostics_portal=onboard_to_aignostics_portal,
                 validate_only=validate_only,
+                gpu_type=gpu_type,
+                gpu_provisioning_mode=gpu_provisioning_mode,
+                max_gpus_per_slide=max_gpus_per_slide,
+                cpu_provisioning_mode=cpu_provisioning_mode,
             )
             logger.debug(
                 "Submitted application run with items: {}, application run id {}, custom metadata: {}",
@@ -941,7 +953,7 @@ class Service(BaseService):  # noqa: PLR0904
             logger.exception(message)
             raise RuntimeError(message) from e
 
-    def application_run_submit(  # noqa: PLR0913, PLR0917
+    def application_run_submit(  # noqa: PLR0913, PLR0917, PLR0912, C901
         self,
         application_id: str,
         items: list[InputItem],
@@ -953,6 +965,10 @@ class Service(BaseService):  # noqa: PLR0904
         deadline: str | None = None,
         onboard_to_aignostics_portal: bool = False,
         validate_only: bool = False,
+        gpu_type: str | None = None,
+        gpu_provisioning_mode: str | None = None,
+        max_gpus_per_slide: int | None = None,
+        cpu_provisioning_mode: str | None = None,
     ) -> Run:
         """Submit a run for the given application.
 
@@ -970,6 +986,10 @@ class Service(BaseService):  # noqa: PLR0904
                 If processing exceeds this deadline, the run can be aborted.
             onboard_to_aignostics_portal (bool): True if the run should be onboarded to the Aignostics Portal.
             validate_only (bool): If True, cancel the run post validation, before analysis.
+            gpu_type (str | None): The type of GPU to use (L4 or A100).
+            gpu_provisioning_mode (str | None): The provisioning mode for GPU resources (SPOT or ON_DEMAND).
+            max_gpus_per_slide (int | None): The maximum number of GPUs to allocate per slide.
+            cpu_provisioning_mode (str | None): The provisioning mode for CPU resources (SPOT or ON_DEMAND).
 
         Returns:
             Run: The submitted run.
@@ -1005,6 +1025,30 @@ class Service(BaseService):  # noqa: PLR0904
                 if deadline:
                     sdk_metadata["scheduling"]["deadline"] = deadline
 
+            if gpu_type or gpu_provisioning_mode or max_gpus_per_slide or cpu_provisioning_mode:
+                sdk_metadata["pipeline"] = {}
+                if gpu_type or gpu_provisioning_mode or max_gpus_per_slide:
+                    sdk_metadata["pipeline"]["gpu"] = {}
+                    if gpu_type:
+                        sdk_metadata["pipeline"]["gpu"]["gpu_type"] = gpu_type
+                    if gpu_provisioning_mode:
+                        sdk_metadata["pipeline"]["gpu"]["provisioning_mode"] = gpu_provisioning_mode
+                    if max_gpus_per_slide:
+                        sdk_metadata["pipeline"]["gpu"]["max_gpus_per_slide"] = max_gpus_per_slide
+                if cpu_provisioning_mode:
+                    sdk_metadata["pipeline"]["cpu"] = {"provisioning_mode": cpu_provisioning_mode}
+
+            # Validate pipeline configuration if present
+            if "pipeline" in sdk_metadata:
+                from aignostics.platform._sdk_metadata import PipelineConfig  # noqa: PLC0415
+
+                try:
+                    PipelineConfig.model_validate(sdk_metadata["pipeline"])
+                except Exception as e:
+                    message = f"Invalid pipeline configuration: {e}"
+                    logger.warning(message)
+                    raise ValueError(message) from e
+
             custom_metadata["sdk"] = sdk_metadata
 
             return self._get_platform_client().runs.submit(

{aignostics-0.2.205 → aignostics-0.2.207}/src/aignostics/platform/_sdk_metadata.py

@@ -11,7 +11,7 @@ from enum import StrEnum
 from typing import Any, Literal
 
 from loguru import logger
-from pydantic import BaseModel, Field, PositiveInt, ValidationError
+from pydantic import BaseModel, Field, ValidationError
 
 from aignostics.utils import user_agent
 
@@ -60,9 +60,11 @@ class GPUConfig(BaseModel):
         default_factory=lambda: ProvisioningMode(DEFAULT_GPU_PROVISIONING_MODE),
         description="The provisioning mode for GPU resources (SPOT or ON_DEMAND)",
     )
-    max_gpus_per_slide: PositiveInt = Field(
+    max_gpus_per_slide: int = Field(
         default=DEFAULT_MAX_GPUS_PER_SLIDE,
-        description="The maximum number of GPUs to allocate per slide",
+        ge=1,
+        le=8,
+        description="The maximum number of GPUs to allocate per slide (1-8)",
     )