yellowdog-sdk 9.1.4__py3-none-any.whl → 9.2.1__py3-none-any.whl

yellowdog_client/_version.py

@@ -1 +1 @@
-__version__ = '9.1.4' # YEL-13378
+__version__ = '9.2.1' # YEL-13449

yellowdog_client/account/account_service_proxy.py

@@ -36,7 +36,7 @@ class AccountServiceProxy:
         return self.proxy.post(AddApplicationResponse, request, "applications")
 
     def regenerate_application_api_key(self, application_id: str) -> ApiKey:
-        return self.proxy.post(url="applications/%s/key" % application_id)
+        return self.proxy.post(ApiKey, url="applications/%s/key" % application_id)
 
     def search_applications(self, search: ApplicationSearch, slice_reference: SliceReference) -> Slice[Application]:
         return self.proxy.get(

yellowdog_client/model/node_worker_target.py

@@ -1,18 +1,24 @@
 from dataclasses import dataclass
+from typing import Optional
 
 from .node_worker_target_type import NodeWorkerTargetType
 
 
 @dataclass
 class NodeWorkerTarget:
-    targetCount: float
     targetType: NodeWorkerTargetType
+    targetCount: Optional[float] = None
+    customTargetCommand: Optional[str] = None
 
     # KEEP
     @staticmethod
     def per_node(target_count: int) -> 'NodeWorkerTarget':
-        return NodeWorkerTarget(target_count, NodeWorkerTargetType.PER_NODE)
+        return NodeWorkerTarget(NodeWorkerTargetType.PER_NODE, target_count)
 
     @staticmethod
     def per_vcpus(target_count: float) -> 'NodeWorkerTarget':
-        return NodeWorkerTarget(target_count, NodeWorkerTargetType.PER_VCPU)
+        return NodeWorkerTarget(NodeWorkerTargetType.PER_VCPU, target_count)
+
+    @staticmethod
+    def per_custom_command(custom_command: str) -> 'NodeWorkerTarget':
+        return NodeWorkerTarget(NodeWorkerTargetType.CUSTOM, None, custom_command)

yellowdog_client/model/node_worker_target_type.py

@@ -4,6 +4,7 @@ from enum import Enum
 class NodeWorkerTargetType(Enum):
     PER_NODE = "PER_NODE"
     PER_VCPU = "PER_VCPU"
+    CUSTOM = "CUSTOM"
 
     def __str__(self) -> str:
         return self.name

yellowdog_client/scheduler/work_client.py

@@ -1,13 +1,12 @@
 from __future__ import annotations
 
 from abc import ABC, abstractmethod
-from typing import List
+from typing import List, Optional
 
 from .work_requirement_helper import WorkRequirementHelper
 from yellowdog_client.common import Closeable, SearchClient
 from yellowdog_client.common.server_sent_events import SubscriptionEventListener
-from yellowdog_client.model import Slice, SliceReference, Task, TaskGroup, TaskSearch, WorkRequirement, \
-    WorkRequirementSearch, WorkRequirementSummary
+from yellowdog_client.model import Slice, SliceReference, Task, TaskGroup, TaskSearch, WorkRequirement, WorkRequirementSearch, WorkRequirementSummary
 
 
 class WorkClient(ABC, Closeable):
@@ -114,32 +113,31 @@ class WorkClient(ABC, Closeable):
         pass
 
     @abstractmethod
-    def cancel_work_requirement(self, work_requirement: WorkRequirement, abort: bool = False) -> WorkRequirement:
+    def cancel_work_requirement(self, work_requirement: WorkRequirement, abort: Optional[bool] = None) -> WorkRequirement:
         """
         Instructs the Scheduler to cancel the supplied work requirement, no further tasks will be executed and all workers shall be released.
 
         :param work_requirement: the work requirement to cancel
-        :param abort: if tasks should be aborted if they have been allocated to a worker
+        :param abort:  if tasks should be aborted if they have been allocated to a worker
         :return: the latest state of the work requirement after the cancel instruction was submitted
         """
 
         pass
 
     @abstractmethod
-    def cancel_work_requirement_by_id(self, work_requirement_id: str, abort: bool = False) -> WorkRequirement:
+    def cancel_work_requirement_by_id(self, work_requirement_id: str, abort: Optional[bool] = None) -> WorkRequirement:
         """
         Instructs the Scheduler to cancel the supplied work requirement, no further tasks will be executed and all workers shall be released.
 
         :param work_requirement_id: the ID of the work requirement to cancel
-        :param abort: if tasks should be aborted if they have been allocated to a worker
+        :param abort:  if tasks should be aborted if they have been allocated to a worker
         :return: the latest state of the work requirement after the cancel instruction was submitted
         """
 
         pass
 
     @abstractmethod
-    def add_work_requirement_listener(self, work_requirement: WorkRequirement,
-                                      listener: SubscriptionEventListener[WorkRequirement]) -> None:
+    def add_work_requirement_listener(self, work_requirement: WorkRequirement, listener: SubscriptionEventListener[WorkRequirement]) -> None:
         """
         Adds an event listener to receive notifications of changes for the specified work requirement.
         The client manages subscriptions to YellowDog Scheduler such that the first listener created for a requirement will cause a Server-Sent Events subscription to be initiated; additional listeners for the same requirement share that subscription.
@@ -151,8 +149,7 @@ class WorkClient(ABC, Closeable):
         pass
 
     @abstractmethod
-    def add_work_requirement_listener_by_id(self, work_requirement_id: str,
-                                            listener: SubscriptionEventListener[WorkRequirement]) -> None:
+    def add_work_requirement_listener_by_id(self, work_requirement_id: str, listener: SubscriptionEventListener[WorkRequirement]) -> None:
         """
         Adds an event listener to receive notifications of changes for the specified work requirement.
         The client manages subscriptions to YellowDog Scheduler such that the first listener created for a requirement will cause a Server-Sent Events subscription to be initiated; additional listeners for the same requirement share that subscription.
@@ -197,13 +194,12 @@ class WorkClient(ABC, Closeable):
         pass
 
     @abstractmethod
-
     def find_all_work_requirements(self) -> List[WorkRequirementSummary]:
         """
         Returns summaries of all existing work requirements within the system for the requesting user.
 
         :return: a list of work requirement summaries
-        @deprecated use {@link #getWorkRequirements(WorkRequirementSearch)} instead to search tasks.
+        @deprecated use #getWorkRequirements(WorkRequirementSearch) instead to search tasks.
         """
 
         pass
@@ -244,8 +240,7 @@ class WorkClient(ABC, Closeable):
         pass
 
     @abstractmethod
-    def add_tasks_to_task_group_by_name(self, namespace: str, work_requirement_name: str, task_group_name: str,
-                                        tasks: List[Task]) -> List[Task]:
+    def add_tasks_to_task_group_by_name(self, namespace: str, work_requirement_name: str, task_group_name: str, tasks: List[Task]) -> List[Task]:
         """
         Submits NEW tasks to the YellowDog Scheduler service to be added to the specified task group.
 

{yellowdog_sdk-9.1.4.dist-info → yellowdog_sdk-9.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: yellowdog-sdk
-Version: 9.1.4
+Version: 9.2.1
 Summary: SDK for the YellowDog Platform
 Author-email: YellowDog Limited <support@yellowdog.co>
 Project-URL: Homepage, https://yellowdog.co

{yellowdog_sdk-9.1.4.dist-info → yellowdog_sdk-9.2.1.dist-info}/RECORD

@@ -1,12 +1,12 @@
 __init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 yellowdog_client/__init__.py,sha256=xHGTw5UbjkeEl_hC8_gJCacfji6462qJWD1nvJdFssE,13162
-yellowdog_client/_version.py,sha256=US9w_fiCtXDv8XAOOa8PKRWIKfxVYpZfB3aD9B_aH8w,34
+yellowdog_client/_version.py,sha256=kXbYBKK3fsu2Y5Ia03tc1H0csyeeQXRntlsAePQjSuY,34
 yellowdog_client/client_collection.py,sha256=VSEzjf6iR1qCQ0YGLyDq_Kgvw8r832QDwTp6-MLB4Vs,388
 yellowdog_client/platform_client.py,sha256=yCzKsOQKllnfzwP9bG-JGtvOdvY7EaEQzFsgX3ERzVY,7476
 yellowdog_client/account/__init__.py,sha256=DiLL3uSMyVlAKWsncX0k5Ioc2hw87HQoEkSYO0ro0fg,456
 yellowdog_client/account/account_client.py,sha256=TRuWPcY4pyk78WXiLU_GOLTSHDhQcpppAGmzFlIPEXw,3398
 yellowdog_client/account/account_client_impl.py,sha256=DUtybxZJpWkS8iyzidJ-veYb9V0iNmEKzB47TcrzzIc,6021
-yellowdog_client/account/account_service_proxy.py,sha256=jHOsXRyNUKnoGYILoN-i1xIOmtpVu4OCK7IkoC8kkGY,5388
+yellowdog_client/account/account_service_proxy.py,sha256=fO1XImrdEhY_uxSQWR8Y7T7z7TQkyfne-RESWy3CuC0,5396
 yellowdog_client/account/keyring_client.py,sha256=IFogh3LNNtL0cSbvbaEuhJ41UhtJbItyxm0AEnAaFMk,1125
 yellowdog_client/account/keyring_client_impl.py,sha256=s_IltFtozR7lYdR9XNEZVj885pgJ0DoInv0B24GtblI,1665
 yellowdog_client/account/keyring_service_proxy.py,sha256=8oK6lK3FxHOIbq--qcKSAd5bqkPGMN023CrX_Q8RKeI,1220
@@ -254,8 +254,8 @@ yellowdog_client/model/node_slot_numbering.py,sha256=qKA4nXwu1iv58rdqaIr9sJhDh_K
 yellowdog_client/model/node_status.py,sha256=25PUpqyPruzc0Y1Rj7RbAJrMLjNKQWDUwrc_0tP1KJw,832
 yellowdog_client/model/node_summary.py,sha256=u5X27f0MxyJb6e24Czb0f7aMkLDUlHdtglQdyB6x214,604
 yellowdog_client/model/node_type.py,sha256=Lh_73zwZz6lKFHyn3chJZmretYuhkhHgzf0H1XNLJ7I,323
-yellowdog_client/model/node_worker_target.py,sha256=1PYMboD2Rz7hTOZ7A1vy-iykXniJOqRGwc-Xamm5rys,513
-yellowdog_client/model/node_worker_target_type.py,sha256=KpsfvAU0ciW7zl8U0vAcnbUPKFxS7Z1cIpTuBxEMDVM,166
+yellowdog_client/model/node_worker_target.py,sha256=uVcU-02dOZZUSqW43c4hf4FKnuw3iQfxs8DabOEJEm4,777
+yellowdog_client/model/node_worker_target_type.py,sha256=z8H_Sk4aTzSxayGf10X9Gi_sCTPCWhSscChVWyFzyo8,188
 yellowdog_client/model/node_write_file_action.py,sha256=OUZrBBT0JMCoSejev1sYsXGqXb2xkUOYkcOD7oTTmcg,401
 yellowdog_client/model/numeric_attribute_constraint.py,sha256=4ijy1MaoDScEi36SnEV_vAq5LixICME6nTXWs6G5QGo,375
 yellowdog_client/model/numeric_attribute_definition.py,sha256=_wPXsHmxW1mhFrbmbtVHzomS7KDhzRmUa0YAaltGaXE,664
@@ -448,7 +448,7 @@ yellowdog_client/object_store/utils/memory_mapped_file_writter.py,sha256=8l9E_m8
 yellowdog_client/scheduler/__init__.py,sha256=ZORtc_LUBoTcplpoDhCHZNannI3qjeohghNcoBYU3ts,927
 yellowdog_client/scheduler/predicated_work_subscription_event_listener.py,sha256=mLfaVRmIMtQbj9u6d8O_B-VaZxOy5l1kuGCfTsS_xPU,2056
 yellowdog_client/scheduler/predicated_worker_pool_subscription_event_listener.py,sha256=z7tCniTF8WL6HSV7QcMOWFvC9EGSRopBGLODFOQinTA,2080
-yellowdog_client/scheduler/work_client.py,sha256=9c2FARQ21JBVsOHDuiYEPs8w_6pM1psYV1G0bUpn43g,13142
+yellowdog_client/scheduler/work_client.py,sha256=_IRRPJLyTI2b8Y65weEEp_o2_GHy_FJAS1OMftuwSso,13035
 yellowdog_client/scheduler/work_client_impl.py,sha256=33O8T25bV-zHx3R3gSQ047NGuSmpWESn7sCaZmqgLFo,7343
 yellowdog_client/scheduler/work_requirement_helper.py,sha256=vQMoI9emTMKLLmZSil3IGNUtxUl50MFuITJTv5Okvlo,4167
 yellowdog_client/scheduler/work_service_proxy.py,sha256=bSg0892B82uAN3zXSUGoSF0BjVzhTuzuZNEMS-wZRIw,2995
@@ -460,8 +460,8 @@ yellowdog_client/usage/__init__.py,sha256=XQwRJqTdxKZa1QUTsxBEL0TqQJeQHGyPklFeqc
 yellowdog_client/usage/allowances_client.py,sha256=H6n63jXjT4OwuWJgFUXSjSmvGTZz9uspy3kj3upinaA,1337
 yellowdog_client/usage/allowances_client_impl.py,sha256=nQPnSzJKhL3WvyCn5fmiDkwE84xZryH9YvV5Z1GjU4M,2061
 yellowdog_client/usage/allowances_service_proxy.py,sha256=uO6LWnpjIzUcZTGdOxPXn7SyYX7NMRqO5KUiHUGr490,1320
-yellowdog_sdk-9.1.4.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-yellowdog_sdk-9.1.4.dist-info/METADATA,sha256=9TXYc5OO3HCDKtzDcak9zWRq_o9IzZky2NW6BbuMSCA,3238
-yellowdog_sdk-9.1.4.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-yellowdog_sdk-9.1.4.dist-info/top_level.txt,sha256=6PH16DcoqpYHhQ5A0UJOjf0tg-1rTrNC9C2CLqCMuFo,26
-yellowdog_sdk-9.1.4.dist-info/RECORD,,
+yellowdog_sdk-9.2.1.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+yellowdog_sdk-9.2.1.dist-info/METADATA,sha256=lgctmkX9IK68IUcQU7V1ijfCiWi_byUB8pvl43COVfU,3238
+yellowdog_sdk-9.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+yellowdog_sdk-9.2.1.dist-info/top_level.txt,sha256=6PH16DcoqpYHhQ5A0UJOjf0tg-1rTrNC9C2CLqCMuFo,26
+yellowdog_sdk-9.2.1.dist-info/RECORD,,

{yellowdog_sdk-9.1.4.dist-info → yellowdog_sdk-9.2.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (79.0.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any