runnable 0.28.1__tar.gz → 0.28.3__tar.gz

{runnable-0.28.1 → runnable-0.28.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: runnable
-Version: 0.28.1
+Version: 0.28.3
 Summary: Add your description here
 Author-email: "Vammi, Vijay" <vijay.vammi@astrazeneca.com>
 License-File: LICENSE

{runnable-0.28.1 → runnable-0.28.3}/extensions/job_executor/k8s.py

@@ -34,6 +34,19 @@ class ImagePullPolicy(str, Enum):
     NEVER = "Never"
 
 
+class TolerationOperator(str, Enum):
+    EXISTS = "Exists"
+    EQUAL = "Equal"
+
+
+class Toleration(BaseModel):
+    key: str
+    operator: TolerationOperator = TolerationOperator.EQUAL
+    value: Optional[str]
+    effect: str
+    toleration_seconds: Optional[int] = Field(default=None)
+
+
 class LabelSelectorRequirement(BaseModel):
     key: str
     operator: Operator
@@ -56,6 +69,12 @@ class EnvVar(BaseModel):
     value: str
 
 
+VendorGPU = Annotated[
+    Optional[int],
+    PlainSerializer(lambda x: str(x), return_type=str, when_used="unless-none"),
+]
+
+
 class Request(BaseModel):
     """
     The default requests
@@ -63,19 +82,16 @@ class Request(BaseModel):
 
     memory: str = "1Gi"
     cpu: str = "250m"
+    gpu: VendorGPU = Field(default=None, serialization_alias="nvidia.com/gpu")
 
 
-VendorGPU = Annotated[
-    Optional[int],
-    PlainSerializer(lambda x: str(x), return_type=str, when_used="unless-none"),
-]
-
-
-class Limit(Request):
+class Limit(BaseModel):
     """
     The default limits
     """
 
+    memory: str = "1Gi"
+    cpu: str = "250m"
     gpu: VendorGPU = Field(default=None, serialization_alias="nvidia.com/gpu")
 
 
@@ -124,7 +140,7 @@ class PVCVolume(BaseModel):
 class K8sTemplateSpec(BaseModel):
     active_deadline_seconds: int = Field(default=60 * 60 * 2)  # 2 hours
     node_selector: Optional[dict[str, str]] = None
-    tolerations: Optional[list[dict[str, str]]] = None
+    tolerations: Optional[list[Toleration]] = None
     volumes: Optional[list[HostPathVolume | PVCVolume]] = Field(
         default_factory=lambda: []
     )
@@ -260,7 +276,7 @@ class GenericK8sJobExecutor(GenericJobExecutor):
         tolerations = None
         if self.job_spec.template.spec.tolerations:
             tolerations = [
-                self._client.V1Toleration(**toleration)
+                self._client.V1Toleration(**toleration.model_dump(by_alias=True))
                 for toleration in self.job_spec.template.spec.tolerations
             ]
 

{runnable-0.28.1 → runnable-0.28.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "runnable"
-version = "0.28.1"
+version = "0.28.3"
 description = "Add your description here"
 readme = "README.md"
 authors = [