yellowdog-sdk 11.2.0__py3-none-any.whl → 11.3.0__py3-none-any.whl

yellowdog_client/_version.py

@@ -1 +1 @@
-__version__ = '11.2.0' # YEL-13439
+__version__ = '11.3.0' # YEL-13745

yellowdog_client/common/proxy.py

@@ -1,12 +1,14 @@
 from dataclasses import dataclass, asdict
+from datetime import datetime, timedelta
 from typing import TypeVar, Type, Dict, Optional
 
 from requests import Request, Session, Response, HTTPError
 from requests.adapters import HTTPAdapter
 from urllib3.util.retry import Retry
-from yellowdog_client.model.exceptions import BaseCustomException
 
+from yellowdog_client.model.exceptions import BaseCustomException
 from .credentials import ApiKeyAuthenticationHeadersProvider
+from .iso_datetime import iso_timedelta_format, iso_format
 from .json import Json
 from .server_sent_events.sse4python import EventSource
 from .user_agent import UserAgent
@@ -56,6 +58,12 @@ class Proxy:
 
     def execute(self, method: str, url: str = "", data: object = None, return_type: Type[T] = None,
                 params: Dict[str, object] = None) -> T:
+        if params is not None:
+            for key, value in params.items():
+                if isinstance(value, datetime):
+                    params[key] = iso_format(value)
+                elif isinstance(value, timedelta):
+                    params[key] = iso_timedelta_format(value)
         response = self.raw_execute(method, url, Json.dump(data) if data else None, params)
         return Json.load(response.json(), return_type) if return_type else None
 

yellowdog_client/model/task_group_status.py

@@ -8,33 +8,36 @@ class TaskGroupStatus(Enum):
     The status of the task group provides an aggregated view of the statuses of tasks within the task group.
     """
 
-    PENDING = "PENDING", False, False
+    PENDING = "PENDING", False, False, False
     """The task group is awaiting resources required to execute tasks."""
-    RUNNING = "RUNNING", False, False
+    RUNNING = "RUNNING", False, False, True
     """The task group has sufficient resources to execute tasks."""
-    HELD = "HELD", False, False
+    HELD = "HELD", False, False, False
     """
     The task group parent work requirement has been held by the user such that no further tasks are executed.
     Resources (e.g. Workers) will be released.
     The task group will remain in HELD state until the user reactivates the parent work requirement.
     """
 
-    COMPLETED = "COMPLETED", False, True
+    FINISHING = "FINISHING", True, False, True
+    """The task group is waiting for all tasks to finish, no further tasks can be added."""
+    COMPLETED = "COMPLETED", False, True, False
     """All tasks within the task group have been completed."""
-    FAILING = "FAILING", True, False
+    FAILING = "FAILING", True, False, False
     """At least one task in the task group has failed and the task group is in the process of discarding any outstanding tasks."""
-    FAILED = "FAILED", False, True
+    FAILED = "FAILED", False, True, False
     """All tasks within the task group have been finished but at least one has failed."""
-    CANCELLING = "CANCELLING", True, False
+    CANCELLING = "CANCELLING", True, False, False
     """The parent work requirement is in the process of being cancelled, no further tasks will be executed."""
-    CANCELLED = "CANCELLED", False, True
+    CANCELLED = "CANCELLED", False, True, False
     """The parent work requirement has been cancelled, no tasks are currently being executed or will be executed."""
 
-    def __new__(cls, value, finishing: bool, finished: bool):
+    def __new__(cls, value, finishing: bool, finished: bool, active: bool):
         obj = object.__new__(cls)
         obj._value_ = value
         obj.finishing = finishing
         obj.finished = finished
+        obj.active = active
         return obj
 
     def __str__(self) -> str:

yellowdog_client/model/work_requirement_status.py

@@ -16,6 +16,8 @@ class WorkRequirementStatus(Enum):
     Task group resources (e.g. Workers) will be released.
     """
 
+    FINISHING = "FINISHING", False
+    """The work requirement is waiting for all tasks to finish, no further tasks can be added."""
     COMPLETED = "COMPLETED", True
     """All task groups in the work requirement have been completed."""
     FAILED = "FAILED", True

yellowdog_client/scheduler/work_client.py

@@ -112,6 +112,28 @@ class WorkClient(ABC, Closeable):
 
         pass
 
+    @abstractmethod
+    def finish_work_requirement(self, work_requirement: WorkRequirement) -> WorkRequirement:
+        """
+        Instructs the Scheduler not to accept further tasks for the supplied work requirement and to finish it after all tasks are finished.
+
+        :param work_requirement: the work requirement to finish
+        :return: the latest state of the work requirement after the finish instruction was submitted
+        """
+
+        pass
+
+    @abstractmethod
+    def finish_work_requirement_by_id(self, work_requirement_id: str) -> WorkRequirement:
+        """
+        Instructs the Scheduler not to accept further tasks for the supplied work requirement and to finish it after all tasks are finished.
+
+        :param work_requirement_id: the ID of the work requirement to finish
+        :return: the latest state of the work requirement after the finish instruction was submitted
+        """
+
+        pass
+
     @abstractmethod
     def cancel_work_requirement(self, work_requirement: WorkRequirement, abort: Optional[bool] = None) -> WorkRequirement:
         """

yellowdog_client/scheduler/work_client_impl.py

@@ -66,6 +66,12 @@ class WorkClientImpl(WorkClient):
     def start_work_requirement_by_id(self, work_requirement_id: str) -> WorkRequirement:
         return self.__service_proxy.transition_work_requirement(work_requirement_id, WorkRequirementStatus.RUNNING, False)
 
+    def finish_work_requirement(self, work_requirement: WorkRequirement) -> WorkRequirement:
+        return self.finish_work_requirement_by_id(work_requirement.id)
+
+    def finish_work_requirement_by_id(self, work_requirement_id: str) -> WorkRequirement:
+        return self.__service_proxy.transition_work_requirement(work_requirement_id, WorkRequirementStatus.FINISHING, False)
+
     def cancel_work_requirement(self, work_requirement: WorkRequirement, abort: bool = False) -> WorkRequirement:
         return self.cancel_work_requirement_by_id(work_requirement.id, abort)
 

{yellowdog_sdk-11.2.0.dist-info → yellowdog_sdk-11.3.0.dist-info}/METADATA

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

{yellowdog_sdk-11.2.0.dist-info → yellowdog_sdk-11.3.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 __init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 yellowdog_client/__init__.py,sha256=xHGTw5UbjkeEl_hC8_gJCacfji6462qJWD1nvJdFssE,13162
-yellowdog_client/_version.py,sha256=oPQnQh-UkT7GrfQR1QGShIlgZrP1oiLuHqkxGWPEPMo,35
+yellowdog_client/_version.py,sha256=CXS8SfCLeWDtc1vsvubY9D5OS92b3HqcVkvLSKa_hJk,35
 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
@@ -19,7 +19,7 @@ yellowdog_client/common/closeable.py,sha256=TLRGcSWDTCQnuhVxXlPyI9BoWkmETyy9h_J2
 yellowdog_client/common/countdown_event.py,sha256=RVGuUmNQ4dSGL3TTOvrNf9A-o9qwJPLZWuYuerQj2ww,1034
 yellowdog_client/common/iso_datetime.py,sha256=Ctnm5paxH8vw0NZBvvaol0L2txgldKEW2leSDH1WyQs,640
 yellowdog_client/common/pagination.py,sha256=ISzq9TkkHEIOA-8XM1MCprbLJzkvOlKCEEVHKZaehT4,540
-yellowdog_client/common/proxy.py,sha256=fEfpT4YMmTxAdRQJxIMUQ8xby-hSiVbyGfd0Prs51vI,8509
+yellowdog_client/common/proxy.py,sha256=0_uD7_ejXFGMydCFU7IArOsaRkErH0YtusV8p0mFu2A,8899
 yellowdog_client/common/search_client.py,sha256=CThwUhlV6g_IrCOBgoyUCjgNhMXKroqGlxdkm-YimQs,1420
 yellowdog_client/common/slice_iterator.py,sha256=Zvp6Vp4jCqet7i3m7A8SK0MYcaQWcio3nMnkQQvsH-I,1303
 yellowdog_client/common/synchronized_predicated_runner.py,sha256=b5g77hWVkbh9iJF3MM6qgYN6y8SxmVmx6hKFDjG20FQ,459
@@ -330,7 +330,7 @@ yellowdog_client/model/task_error.py,sha256=9aW_ZJ0WkQjUSbpasXBtc0Wj4iu4-70Mf3jt
 yellowdog_client/model/task_error_matcher.py,sha256=idOfRZllKDlqe8Wq2GMjEaYhLLrAmd7Hai5Dr5BJMd0,291
 yellowdog_client/model/task_error_type.py,sha256=3JDxDu8_f_eQdoB_Y1xuVW4N_leSoJAGn5OjcU4X3e0,701
 yellowdog_client/model/task_group.py,sha256=29t6oxZda_P0VtnUlzQn0o8An38cD0B4jbwGiVKIqCE,2321
-yellowdog_client/model/task_group_status.py,sha256=7wAEOvZzq07P_0vTP0NejmDLS-MifGWO5WLlu_5OdZg,1722
+yellowdog_client/model/task_group_status.py,sha256=dNFk111vk3iTOW0mPjRwUrTn_5yTbgngmJ8Rc8jlElM,1958
 yellowdog_client/model/task_input.py,sha256=d3U32xhgHdEWtSu5uSjjsLniD9dXoNeIVsyl1BeZqlw,1975
 yellowdog_client/model/task_input_source.py,sha256=Zr_p6dT0WMdMCcS2boGxPunFIa8CJB9hFguJwV85-IA,423
 yellowdog_client/model/task_input_verification.py,sha256=NYWsdtW4anEjGlqeta3mDpDPL9XmQj-0QBvb1n41svM,477
@@ -357,7 +357,7 @@ yellowdog_client/model/waterfall_provision_strategy.py,sha256=1mAIACbWcVrsy3Bjul
 yellowdog_client/model/work_requirement.py,sha256=4-kqhjS33HEjf62Z_GjcgVxHZ4Xa5c54TR1DIDgYPuY,1714
 yellowdog_client/model/work_requirement_dashboard_summary.py,sha256=BeiBs92lS1InDuYc4yt5uuWKhoNqlfdXkk3PmyinIuc,273
 yellowdog_client/model/work_requirement_search.py,sha256=xQ6U7v7O0udd2LB79DjLzX4DqpuifIwzip7Uv17jU7Y,621
-yellowdog_client/model/work_requirement_status.py,sha256=w_XCB-SDBJQjHnkM28UivdA53MVaAlDk0PzUcZyZYmk,1301
+yellowdog_client/model/work_requirement_status.py,sha256=JXFDJLakCmpUwv1R6j97lfU_YN7Ppb4YRvhYtWqclkA,1434
 yellowdog_client/model/work_requirement_summary.py,sha256=jEvsqI6IBykpP5_LAyoTiS5elHXfaQJ2Zj1FOwxQtMU,1374
 yellowdog_client/model/worker.py,sha256=KqUNIYwy9F-R4eY3Vk2JkZIyhERE-2CxmQe-6JgkI5Q,964
 yellowdog_client/model/worker_action.py,sha256=58HCgRGrOVIsz3aKPdwGrY51w5LQQwFlVPKBQ6808Wg,520
@@ -453,8 +453,8 @@ 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=_IRRPJLyTI2b8Y65weEEp_o2_GHy_FJAS1OMftuwSso,13035
-yellowdog_client/scheduler/work_client_impl.py,sha256=33O8T25bV-zHx3R3gSQ047NGuSmpWESn7sCaZmqgLFo,7343
+yellowdog_client/scheduler/work_client.py,sha256=psfLSrJoJSNtj3tGD7lRHIZsLOTmUeaP7wdvdXGolLA,13963
+yellowdog_client/scheduler/work_client_impl.py,sha256=oGabaKD5vU5UzrqTTtwKlnsVXTWx8xFLheWHTm_eGkw,7724
 yellowdog_client/scheduler/work_requirement_helper.py,sha256=vQMoI9emTMKLLmZSil3IGNUtxUl50MFuITJTv5Okvlo,4167
 yellowdog_client/scheduler/work_service_proxy.py,sha256=bSg0892B82uAN3zXSUGoSF0BjVzhTuzuZNEMS-wZRIw,2995
 yellowdog_client/scheduler/worker_pool_client.py,sha256=31JATKHyDTCCIOGVQGKLB_4S-u24EwVWDsYe_K1genY,15723
@@ -465,8 +465,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-11.2.0.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-yellowdog_sdk-11.2.0.dist-info/METADATA,sha256=t-q0tR_wvKo_2gzqcPKXPmboTaR7wJQYmwWwr_oeF50,3239
-yellowdog_sdk-11.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-yellowdog_sdk-11.2.0.dist-info/top_level.txt,sha256=6PH16DcoqpYHhQ5A0UJOjf0tg-1rTrNC9C2CLqCMuFo,26
-yellowdog_sdk-11.2.0.dist-info/RECORD,,
+yellowdog_sdk-11.3.0.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+yellowdog_sdk-11.3.0.dist-info/METADATA,sha256=TQDZiVhJNeslKlUV1Hk5zfexGjdFsB-GDsJVD5zy1II,3239
+yellowdog_sdk-11.3.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+yellowdog_sdk-11.3.0.dist-info/top_level.txt,sha256=6PH16DcoqpYHhQ5A0UJOjf0tg-1rTrNC9C2CLqCMuFo,26
+yellowdog_sdk-11.3.0.dist-info/RECORD,,