arkitekt-next 0.7.18__tar.gz → 0.7.19__tar.gz

{arkitekt_next-0.7.18 → arkitekt_next-0.7.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arkitekt-next
-Version: 0.7.18
+Version: 0.7.19
 Summary: client for the arkitekt_next platform
 License: MIT
 Author: jhnnsrs

{arkitekt_next-0.7.18 → arkitekt_next-0.7.19}/arkitekt_next/cli/commands/init/main.py

@@ -125,14 +125,6 @@ def ensure_semver(ctx, param, value):
     default=False,
     callback=check_overwrite_app,
 )
-@click.option(
-    "--requirements",
-    "-r",
-    help="Dedicated hardware requirements for this app. They will be used to inform Plugin Schedulers like port, on which compute resources to put your plugin app, once deployed.You can choose multiple for your app. For a list of requirements, run `arkitekt_next manifest requirements available`",
-    type=click.Choice(compile_requirements()),
-    multiple=True,
-    default=[],
-)
 @click.option(
     "--scopes",
     "-s",

{arkitekt_next-0.7.18 → arkitekt_next-0.7.19}/arkitekt_next/cli/commands/kabinet/build.py

@@ -140,7 +140,7 @@ def inspect_requirements(build_id: str) -> Dict[str, Requirement]:
 
         # Parse the JSON output
         correct_part = result.stdout.split("--START_REQUIREMENTS--")[1].split(
-            "--START_REQUIREMENTS--"
+            "--END_REQUIREMENTS--"
         )[0]
 
         try:
@@ -151,7 +151,7 @@ def inspect_requirements(build_id: str) -> Dict[str, Requirement]:
                 f"Could not decode JSON output of docker inspect. {combined_error}"
             ) from e
 
-        return output.values()
+        return output
     except subprocess.CalledProcessError as e:
         combined_error = e.stdout + e.stderr
 
@@ -167,8 +167,9 @@ def inspect_requirements(build_id: str) -> Dict[str, Requirement]:
 def inspect_build(build_id: str) -> Inspection:
     size, size_root_fs = inspect_docker_container(build_id)
     templates = inspect_templates(build_id)
+    requirements = inspect_requirements(build_id)
 
-    return Inspection(size=size, templates=templates)
+    return Inspection(size=size, templates=templates , requirements=requirements)
 
 
 def get_flavours(ctx: Context, select: Optional[str] = None) -> Dict[str, Flavour]:

{arkitekt_next-0.7.18 → arkitekt_next-0.7.19}/arkitekt_next/cli/constants.py

@@ -9,10 +9,6 @@ def compile_scopes() -> List[str]:
     return ["read", "write"]
 
 
-def compile_requirements():
-    """Compile all available requirements"""
-    return [Requirement.GPU.value]
-
 
 def compile_builders():
     return ["arkitekt_next.builders.easy", "arkitekt_next.builders.port"]

{arkitekt_next-0.7.18 → arkitekt_next-0.7.19}/arkitekt_next/cli/types.py

@@ -1,9 +1,10 @@
 from pydantic import BaseModel, Field, validator
 import datetime
-from typing import List, Optional, Union, Literal
+from typing import List, Optional, Union, Literal, Dict
 from enum import Enum
 import semver
 import uuid
+from arkitekt_next.model import Requirement
 from rekuest.api.schema import DefinitionInput
 from string import Formatter
 import os
@@ -24,12 +25,6 @@ class Framework(str, Enum):
     PYTORCH = "pytorch"
 
 
-class Requirement(str, Enum):
-    """ """
-
-    GPU = "gpu"
-
-
 class Packager(str, Enum):
     CONDA = "conda"
     POETRY = "poetry"
@@ -43,7 +38,6 @@ class Manifest(BaseModel):
     logo: Optional[str]
     entrypoint: str
     scopes: List[str]
-    requirements: List[Requirement] = Field(default_factory=list)
     created_at: datetime.datetime = Field(default_factory=datetime.datetime.now)
 
     @validator("version", pre=True)
@@ -218,6 +212,7 @@ Selector = Union[
 
 class Inspection(BaseModel):
     templates: List[TemplateInput]
+    requirements: Dict[str, Requirement]
     size: int
 
 

{arkitekt_next-0.7.18 → arkitekt_next-0.7.19}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "arkitekt-next"
-version = "0.7.18"
+version = "0.7.19"
 description = "client for the arkitekt_next platform"
 authors = ["jhnnsrs <jhnnsrs@gmail.com>"]
 license = "MIT"